ccr-cheng / english-to-vietnamese-nmt-model Goto Github PK

python3 main.py train

optional arguments:
  -h, --help            show this help message and exit
  --lr                  initial learning rate [default: 5e-4]
  --lr-decay            learning rate decay factor [default: 0.5]
  --patience            patience before lr decay [default: 3]
  --num-epoch           number of training epochs [default:20]
  --num-layer           number of layers in encoder and decoder [default: 6]
  --batch-size          batch size [default: 64]
  --embed-size          embedding size (d_model) [default: 256]
  --hidden-size         hidden size (in the feedforward layers) [default: 512]
  --max-length          max length to trim the dataset [default: 64]
  --clip-grad           parameter clipping threshold [default: 1.0]
  --print-every         print training procedure every number of batches [default: 100]
  --save-path           model path for saving  [default: 'NMT.pt']
  --checkpoint          checkpoint for resuming training [default: '']

python3 main.py test --model [model file]