#!/bin/bash
export CUDA_VISIBLE_DEVICES=0

python train.py \
        --src_lang en --tar_lang vi \
        --num_layers 2 \
        --hidden_size 512 \
        --src_vocab_size 17191 \
        --tar_vocab_size 7709 \
        --batch_size 128 \
        --dropout 0.2 \
        --init_scale  0.2 \
        --max_grad_norm 5.0 \
        --train_data_prefix data/en-vi/train \
        --eval_data_prefix data/en-vi/tst2012 \
        --test_data_prefix data/en-vi/tst2013 \
        --vocab_prefix data/en-vi/vocab \
        --use_gpu True \
        --model_path attention_models \
        --enable_ce \
        --learning_rate 0.002 \
        --dtype float32 \
        --optimizer adam \
        --max_epoch 12 \
        --padding_idx 2