File size: 691 Bytes
d105bed |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
{
"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
}
} |