Train the model using GenQ (getting 80% top-5 accuracy)

README.md

@@ -84,7 +84,7 @@ The model was trained with the parameters:
 
 **DataLoader**:
 
-`torch.utils.data.dataloader.DataLoader` of length 170 with parameters:
+`torch.utils.data.dataloader.DataLoader` of length 165 with parameters:
 ```
 {'batch_size': 16, 'sampler': 'torch.utils.data.sampler.SequentialSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
 ```
@@ -110,7 +110,7 @@ Parameters of the fit()-Method:
     },
     "scheduler": "WarmupLinear",
     "steps_per_epoch": null,
-    "warmup_steps": 25,
+    "warmup_steps": 24,
     "weight_decay": 0.01
 }
 ```

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4848c55c2b50d8f6d125ce82cec5bed9753263488ea31952e8fb57825c8be0d9
-size 436634191
+oid sha256:3cba22292dbf47eb0e2163fb93a601f3cd3f27c46ba3632e9ccf4d36c39dec6d
+size 436634195