Update train.py
Browse files
train.py
CHANGED
|
@@ -56,7 +56,7 @@ training_args = TrainingArguments(
|
|
| 56 |
evaluation_strategy="epoch",
|
| 57 |
save_strategy="epoch",
|
| 58 |
push_to_hub=False,
|
| 59 |
-
fp16=
|
| 60 |
)
|
| 61 |
|
| 62 |
# Set up Trainer
|
|
|
|
| 56 |
evaluation_strategy="epoch",
|
| 57 |
save_strategy="epoch",
|
| 58 |
push_to_hub=False,
|
| 59 |
+
fp16=False
|
| 60 |
)
|
| 61 |
|
| 62 |
# Set up Trainer
|