code

viraweb123 · Aug 14, 2024 · e565c0c · e565c0c
1 parent 49d672e
commit e565c0c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/train/code/main.py b/train/code/main.py
@@ -26,6 +26,7 @@
 per_device_train_batch_size_var = 2
 num_train_epochs_var = 50
 save_steps_var = 5000
+logging_steps = 100
 
 logging.info("Starting the training process")
 
@@ -83,7 +84,7 @@ def train(train_file_path,
         per_device_train_batch_size=per_device_train_batch_size,
         num_train_epochs=num_train_epochs,
         logging_dir=logging_dir, 
-        logging_steps=10,
+        logging_steps=logging_steps,
         save_steps=save_steps
     )
     trainer = Trainer(