init

2022-08-07 13:38:07 +09:00
parent c173a6d7dc
commit 655eedc421
3 changed files with 8 additions and 5 deletions
--- a/VitTrainer.py
+++ b/VitTrainer.py
@@ -52,7 +52,7 @@ class VitTrainer:
            metric_for_best_model="accuracy",
            logging_dir='logs',
            remove_unused_columns=False,
-            num_train_epochs=20,
+            num_train_epochs=14,
        )

        return