run.sh 732 字节
Newer Older
1 2 3 4 5 6 7 8 9 10
#!/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 \
L
LiuChiaChi 已提交
11
        --dropout 0.2 \
L
LiuChiaChi 已提交
12
        --init_scale  0.2 \
13 14 15 16 17 18
        --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 \
L
LiuChiaChi 已提交
19 20 21
        --model_path attention_models \
        --enable_ce \
        --learning_rate 0.002 \
L
LiuChiaChi 已提交
22
        --dtype float32 \
L
LiuChiaChi 已提交
23
        --optimizer adam \
L
LiuChiaChi 已提交
24 25
        --max_epoch 12 \
        --padding_idx 2