Spaces:

fioriclass
/

emotion_classifier

Sleeping

fioriclass commited on 13 days ago

Commit

bc3d9ee

1 Parent(s): 024f027

augmentation du batch size

Files changed (1) hide show

src/conf/model.yaml CHANGED Viewed

@@ -15,7 +15,7 @@ models:
     num_labels: 2
     learning_rate: 5e-5
     epochs: 3
-    batch_size: 8
     warmup_steps: 0
     weight_decay: 0.0
     adam_epsilon: 1e-8
@@ -26,7 +26,7 @@ models:
     num_labels: 2
     learning_rate: 5e-5
     epochs: 3
-    batch_size: 8
     warmup_steps: 0
     weight_decay: 0.0
     adam_epsilon: 1e-8

     num_labels: 2
     learning_rate: 5e-5
     epochs: 3
+    batch_size: 12
     warmup_steps: 0
     weight_decay: 0.0
     adam_epsilon: 1e-8
     num_labels: 2
     learning_rate: 5e-5
     epochs: 3
+    batch_size: 12
     warmup_steps: 0
     weight_decay: 0.0
     adam_epsilon: 1e-8