| { | |
| "model_name": "Helsinki-NLP/opus-mt-mul-en", | |
| "direction": "he2yid", | |
| "vocabulary_size": 64174, | |
| "model_parameters": 77519872, | |
| "training_config": { | |
| "dataset_path": "hebrew_yiddish_dataset", | |
| "output_dir": "./hebrew_yiddish_model_improved", | |
| "model_name": "Helsinki-NLP/opus-mt-mul-en", | |
| "batch_size": 8, | |
| "learning_rate": 1e-05, | |
| "num_epochs": 100, | |
| "max_input_length": 512, | |
| "max_target_length": 512, | |
| "eval_steps": 500, | |
| "save_steps": 500, | |
| "warmup_steps": 1000, | |
| "gradient_accumulation_steps": 2, | |
| "use_fp16": true, | |
| "use_wandb": false, | |
| "use_language_prefix": true, | |
| "skip_evaluation": false, | |
| "seed": 42, | |
| "resume_from_checkpoint": "./hebrew_yiddish_model_improved/checkpoint-61500", | |
| "per_device_train_batch_size": 4, | |
| "per_device_eval_batch_size": 8 | |
| } | |
| } |