Saved model checkpoint after epoch 0

.gitattributes

@@ -34,3 +34,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
 model.safetensors filter=lfs diff=lfs merge=lfs -text
+.git/lfs/objects/fb/47/fb47583d9ebb29cc3a89ffd4acaa3c3c7376f14705eda1a9f62f076dd44a9813 filter=lfs diff=lfs merge=lfs -text

README.md

@@ -100,7 +100,7 @@ The model was trained with the parameters:
 Parameters of the fit()-Method:
 ```
 {
-    "epochs": 200,
+    "epochs": 20,
     "evaluator": "dfm_sentence_trf.evaluation.task_evaluator.TaskListEvaluator",
     "max_grad_norm": 1,
     "optimizer_class": "<class 'torch.optim.adamw.AdamW'>",
@@ -108,7 +108,7 @@ Parameters of the fit()-Method:
         "lr": 2e-05
     },
     "scheduler": "WarmupLinear",
-    "steps_per_epoch": 500,
+    "steps_per_epoch": 5000,
     "warmup_steps": 100,
     "weight_decay": 0.01
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:fb47583d9ebb29cc3a89ffd4acaa3c3c7376f14705eda1a9f62f076dd44a9813
+oid sha256:4074bdfa870dd130ef7020cca37ef2afa2e98d99afbf525e0bba156a51f06808
 size 87532520