| { | |
| "model_name": "Helsinki-NLP/opus-mt-mul-en", | |
| "direction": "he2en", | |
| "vocabulary_size": 64174, | |
| "model_parameters": 77519872, | |
| "training_config": { | |
| "dataset_path": "./hebrew_english_dataset", | |
| "output_dir": "./hebrew_english_model", | |
| "model_name": "Helsinki-NLP/opus-mt-mul-en", | |
| "direction": "he2en", | |
| "batch_size": 8, | |
| "learning_rate": 2e-05, | |
| "num_epochs": 20, | |
| "max_input_length": 512, | |
| "max_target_length": 512, | |
| "eval_steps": 500, | |
| "save_steps": 500, | |
| "warmup_steps": 500, | |
| "gradient_accumulation_steps": 1, | |
| "seed": 42, | |
| "use_fp16": true, | |
| "use_wandb": false, | |
| "use_language_prefix": true, | |
| "skip_evaluation": false | |
| } | |
| } |