Create example_train.yml

dhvalden · Mar 30, 2022 · 1be1559 · 1be1559
1 parent eebc49c
commit 1be1559
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/example_train.yml b/example_train.yml
@@ -0,0 +1,9 @@
+HF:
+  model_arch: distilbert
+  model_name: distilbert-base-multilingual-cased
+CONFIG:
+  data_path: data/data.csv
+  out_path: models/model1_distilBERT
+  batch_size: 32
+  epochs: 10
+  test_size: 0.20