Rajut commited on
Commit
f7cac21
·
verified ·
1 Parent(s): 07de6eb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -33,12 +33,12 @@ data_collator = DataCollatorForLanguageModeling(
33
  training_args = TrainingArguments(
34
  output_dir=output_dir,
35
  overwrite_output_dir=True,
36
- num_train_epochs=20,
37
  per_device_train_batch_size=1,
38
  save_steps=10_000,
39
  save_total_limit=2,
40
  logging_dir=output_dir,
41
- logging_steps=50,
42
  report_to=[],
43
  )
44
 
 
33
  training_args = TrainingArguments(
34
  output_dir=output_dir,
35
  overwrite_output_dir=True,
36
+ num_train_epochs=10,
37
  per_device_train_batch_size=1,
38
  save_steps=10_000,
39
  save_total_limit=2,
40
  logging_dir=output_dir,
41
+ logging_steps=100,
42
  report_to=[],
43
  )
44