dummy arg values for quicker train
@@ -117,17 +117,17 @@ class HFTransformersModel(BaseModel):
"dataset_config_name": "tr",
"output_dir": "./wav2vec2-large-xlsr-na-4",
"overwrite_output_dir": True,
- "num_train_epochs": "30",
- "per_device_train_batch_size": "4",
- "per_device_eval_batch_size": "4",
- "gradient_accumulation_steps": "2",
+ "num_train_epochs": "2",
+ "per_device_train_batch_size": "1",
+ "per_device_eval_batch_size": "1",
+ "gradient_accumulation_steps": "1",
"learning_rate": "3e-4",
- "warmup_steps": "500",
+ "warmup_steps": "5",
"evaluation_strategy": "steps",
- "save_steps": "400",
- "eval_steps": "400",
- "logging_steps": "400",
- "save_total_limit": "3",
+ "save_steps": "1",
+ "eval_steps": "1",
+ "logging_steps": "1",
+ "save_total_limit": "1",
"freeze_feature_extractor": True,
"feat_proj_dropout": "0.0",
"layerdrop": "0.1",
