Updating with freshly trained model

.gitattributes

@@ -28,3 +28,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 pytorch_model.bin filter=lfs diff=lfs merge=lfs -text
 .git/lfs/objects/60/ce/60ce79a96d7a41a207e79cfea9d5b62876b04591c19495178bedc5ff7c475311 filter=lfs diff=lfs merge=lfs -text
 .git/lfs/objects/f5/74/f5743ce637a953ee1776d97a870a26a646ffc332a295a4492acc2be2fe54daa3 filter=lfs diff=lfs merge=lfs -text
+.git/lfs/objects/32/58/32585d7934eb60f052d5f63f7c6ba6159183bccc30cde299785c25177df7c884 filter=lfs diff=lfs merge=lfs -text

README.md

@@ -81,7 +81,7 @@ The model was trained with the parameters:
 
 **DataLoader**:
 
-`torch.utils.data.dataloader.DataLoader` of length 285 with parameters:
+`torch.utils.data.dataloader.DataLoader` of length 2841 with parameters:
 ```
 {'batch_size': 16, 'sampler': 'torch.utils.data.sampler.RandomSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
 ```
@@ -97,7 +97,7 @@ Parameters of the fit()-Method:
 ```
 {
     "callback": null,
-    "epochs": 1,
+    "epochs": 5,
     "evaluation_steps": 0,
     "evaluator": "NoneType",
     "max_grad_norm": 1,
@@ -106,7 +106,7 @@ Parameters of the fit()-Method:
         "lr": 2e-05
     },
     "scheduler": "WarmupLinear",
-    "steps_per_epoch": 10,
+    "steps_per_epoch": null,
     "warmup_steps": 10000,
     "weight_decay": 0.01
 }

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:32585d7934eb60f052d5f63f7c6ba6159183bccc30cde299785c25177df7c884
+oid sha256:6c338abd9c702b3be95cec15e23c96cf0d8c2e3b7c45df8d48de354f048e1d00
 size 498669047