code

viraweb123 · Aug 14, 2024 · dd5b7f5 · dd5b7f5
1 parent e630efb
commit dd5b7f5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/train/code/main.py b/train/code/main.py
@@ -26,8 +26,8 @@
 overwrite_output_dir_var = True
 per_device_train_batch_size_var = 10
 num_train_epochs_var = 32
-save_steps_var = 5000
-logging_steps = 100
+save_steps_var = 100
+logging_steps = 5
 
 logging.info("Starting the training process")