epoch and batch sizer changed

viraweb123 · Aug 14, 2024 · 2422c37 · 2422c37
1 parent fde942a
commit 2422c37
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/train/code/main.py b/train/code/main.py
@@ -24,8 +24,8 @@
 train_file_path_var = os.path.join(input_base_dir, 'test.txt')
 model_name_var = 'gpt2'
 overwrite_output_dir_var = True
-per_device_train_batch_size_var = 2
-num_train_epochs_var = 50
+per_device_train_batch_size_var = 100
+num_train_epochs_var = 32
 save_steps_var = 5000
 logging_steps = 100