Views
No views yet
1training_args = Seq2SeqTrainingArguments(
2 evaluation_strategy="epoch",
3 save_strategy="epoch",
4 save_total_limit=2,
5 warmup_steps=1000,
6 learning_rate=5e-5,
7 per_device_train_batch_size=16,
8 per_device_eval_batch_size=16,
9 num_train_epochs=3,
10 weight_decay=0.01,
11 predict_with_generate=True,
12 fp16=True
13)