artefucktor commited on
Commit
936baa6
1 Parent(s): 0c1f489

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -53,9 +53,9 @@ The model was trained with the parameters:
53
 
54
  **DataLoader**:
55
 
56
- `torch.utils.data.dataloader.DataLoader` of length 2376 with parameters:
57
  ```
58
- {'batch_size': 8, 'sampler': 'torch.utils.data.sampler.RandomSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
59
  ```
60
 
61
  **Loss**:
 
53
 
54
  **DataLoader**:
55
 
56
+ `torch.utils.data.dataloader.DataLoader` of length 5585 with parameters:
57
  ```
58
+ {'batch_size': 16, 'sampler': 'torch.utils.data.sampler.RandomSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
59
  ```
60
 
61
  **Loss**: