Update train.py
Browse files
train.py
CHANGED
@@ -55,7 +55,7 @@ training_args = TrainingArguments(
|
|
55 |
logging_steps=50,
|
56 |
evaluation_strategy="epoch",
|
57 |
save_strategy="epoch",
|
58 |
-
push_to_hub=False
|
59 |
fp16=True
|
60 |
)
|
61 |
|
|
|
55 |
logging_steps=50,
|
56 |
evaluation_strategy="epoch",
|
57 |
save_strategy="epoch",
|
58 |
+
push_to_hub=False,
|
59 |
fp16=True
|
60 |
)
|
61 |
|