{ "save_data": "data", "seed": 1234, "tensorboard_log_dir_dated": "tensorboard/Jun-29_20-46-44", "src_vocab": "lv.eole.vocab", "share_vocab": false, "n_sample": 0, "vocab_size_multiple": 8, "report_every": 100, "tensorboard_log_dir": "tensorboard", "src_vocab_size": 32000, "tgt_vocab": "en.eole.vocab", "valid_metrics": [ "BLEU" ], "overwrite": true, "tgt_vocab_size": 32000, "transforms": [ "sentencepiece", "filtertoolong" ], "tensorboard": true, "training": { "batch_size": 6000, "accum_steps": [ 0 ], "max_grad_norm": 0.0, "valid_batch_size": 2048, "compute_dtype": "torch.float16", "optim": "adamw", "keep_checkpoint": 4, "dropout": [ 0.1 ], "average_decay": 0.0001, "bucket_size": 128000, "prefetch_factor": 32, "learning_rate": 2.0, "dropout_steps": [ 0 ], "warmup_steps": 2000, "label_smoothing": 0.1, "adam_beta2": 0.998, "normalization": "tokens", "num_workers": 0, "valid_steps": 5000, "model_path": "quickmt-lv-en-eole-model", "decay_method": "noam", "accum_count": [ 15 ], "save_checkpoint_steps": 5000, "train_steps": 100000, "attention_dropout": [ 0.1 ], "batch_type": "tokens", "param_init_method": "xavier_uniform", "batch_size_multiple": 8, "gpu_ranks": [ 0 ], "world_size": 1 }, "data": { "corpus_1": { "path_src": "train.lv", "transforms": [ "sentencepiece", "filtertoolong" ], "path_align": null, "path_tgt": "train.en" }, "valid": { "path_src": "valid.lv", "transforms": [ "sentencepiece", "filtertoolong" ], "path_align": null, "path_tgt": "valid.en" } }, "model": { "heads": 8, "architecture": "transformer", "position_encoding_type": "SinusoidalInterleaved", "hidden_size": 1024, "transformer_ff": 4096, "share_decoder_embeddings": true, "share_embeddings": false, "decoder": { "layers": 2, "decoder_type": "transformer", "heads": 8, "position_encoding_type": "SinusoidalInterleaved", "hidden_size": 1024, "transformer_ff": 4096, "n_positions": null, "tgt_word_vec_size": 1024 }, "embeddings": { "tgt_word_vec_size": 1024, "word_vec_size": 1024, "src_word_vec_size": 1024, "position_encoding_type": "SinusoidalInterleaved" }, "encoder": { "layers": 8, "heads": 8, "position_encoding_type": "SinusoidalInterleaved", "hidden_size": 1024, "transformer_ff": 4096, "src_word_vec_size": 1024, "n_positions": null, "encoder_type": "transformer" } }, "transforms_configs": { "filtertoolong": { "tgt_seq_length": 256, "src_seq_length": 256 }, "sentencepiece": { "tgt_subword_model": "${MODEL_PATH}/en.spm.model", "src_subword_model": "${MODEL_PATH}/lv.spm.model" } } }