reduce batch size
@@ -165,8 +165,8 @@ class HFTransformersModel(BaseModel):
"output_dir": self.path.joinpath(self.OUTPUT_DIR_NAME),
"overwrite_output_dir": True,
"num_train_epochs": NUM_TRAIN_EPOCHS,
- "per_device_train_batch_size": "32",
- "per_device_eval_batch_size": "32",
+ "per_device_train_batch_size": "16",
+ "per_device_eval_batch_size": "16",
"gradient_accumulation_steps": "2",
"learning_rate": "5e-4",
"weight_decay": "0.005",
