| training_args = TrainingArguments( | |
| output_dir="./results", | |
| learning_rate=5e-5, | |
| per_device_train_batch_size=1, | |
| per_device_eval_batch_size=1, | |
| num_train_epochs=5, | |
| weight_decay=0.01, | |
| evaluation_strategy="epoch", | |
| push_to_hub=True | |
| ) |
| training_args = TrainingArguments( | |
| output_dir="./results", | |
| learning_rate=5e-5, | |
| per_device_train_batch_size=1, | |
| per_device_eval_batch_size=1, | |
| num_train_epochs=5, | |
| weight_decay=0.01, | |
| evaluation_strategy="epoch", | |
| push_to_hub=True | |
| ) |