Update train.py

train.py

@@ -47,7 +47,7 @@ def try_training_with_batch_size(batch_size_start):
                 learning_rate=5e-5,
                 per_device_train_batch_size=batch_size,
                 per_device_eval_batch_size=batch_size,
-                fp16=True,
+                #fp16=True,
                 num_train_epochs=3,
                 logging_steps=100,
                 warmup_ratio=0.06,