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
|