gaurav-mac commited on
Commit
5ebb20f
1 Parent(s): a16d736

Pushing hi sbert using sumo dataset with base l3cubepune

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -85,7 +85,7 @@ The model was trained with the parameters:
85
 
86
  **DataLoader**:
87
 
88
- `torch.utils.data.dataloader.DataLoader` of length 66 with parameters:
89
  ```
90
  {'batch_size': 16, 'sampler': 'torch.utils.data.sampler.RandomSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
91
  ```
@@ -110,7 +110,7 @@ Parameters of the fit()-Method:
110
  },
111
  "scheduler": "WarmupLinear",
112
  "steps_per_epoch": null,
113
- "warmup_steps": 990,
114
  "weight_decay": 0.01
115
  }
116
  ```
 
85
 
86
  **DataLoader**:
87
 
88
+ `torch.utils.data.dataloader.DataLoader` of length 80 with parameters:
89
  ```
90
  {'batch_size': 16, 'sampler': 'torch.utils.data.sampler.RandomSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
91
  ```
 
110
  },
111
  "scheduler": "WarmupLinear",
112
  "steps_per_epoch": null,
113
+ "warmup_steps": 1200,
114
  "weight_decay": 0.01
115
  }
116
  ```