batch

train/code/main.py

@@ -24,7 +24,7 @@
 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 = 30
+per_device_train_batch_size_var = 10
 num_train_epochs_var = 32
 save_steps_var = 5000
 logging_steps = 100