marianmt-he2en-nwt / model_info.json
johnlockejrr's picture
Upload 15 files
d105bed verified
{
"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
}
}