nejox commited on
Commit
fad7044
·
1 Parent(s): 5161908

Pushing fine-tuned SimSearch model

Browse files

Fine-tuned sentence transformer model for coffee

Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -46,7 +46,7 @@ The model was trained with the parameters:
46
 
47
  **DataLoader**:
48
 
49
- `torch.utils.data.dataloader.DataLoader` of length 13 with parameters:
50
  ```
51
  {'batch_size': 8, 'sampler': 'torch.utils.data.sampler.RandomSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
52
  ```
@@ -71,7 +71,7 @@ Parameters of the fit()-Method:
71
  },
72
  "scheduler": "WarmupLinear",
73
  "steps_per_epoch": null,
74
- "warmup_steps": 13,
75
  "weight_decay": 0.01
76
  }
77
  ```
 
46
 
47
  **DataLoader**:
48
 
49
+ `torch.utils.data.dataloader.DataLoader` of length 46 with parameters:
50
  ```
51
  {'batch_size': 8, 'sampler': 'torch.utils.data.sampler.RandomSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
52
  ```
 
71
  },
72
  "scheduler": "WarmupLinear",
73
  "steps_per_epoch": null,
74
+ "warmup_steps": 46,
75
  "weight_decay": 0.01
76
  }
77
  ```