CoderHassan commited on
Commit
ff6b765
·
verified ·
1 Parent(s): 783f9ba

Create training_args.bin

Browse files
Files changed (1) hide show
  1. training_args.bin +11 -0
training_args.bin ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import TrainingArguments
2
+
3
+ training_args = TrainingArguments(
4
+ output_dir='./results',
5
+ evaluation_strategy="epoch",
6
+ learning_rate=2e-5,
7
+ per_device_train_batch_size=8,
8
+ per_device_eval_batch_size=8,
9
+ num_train_epochs=3
10
+ )
11
+ training_args.save('./t5_urdu_translation/training_args.bin')