#!/usr/bin/env bash
export CUDA_VISIBLE_DEVICES=$1
export SRC="uy"
export TGT="zh"
export LAN="${SRC}2${TGT}"
export CON="/home/user_data55/wangdq/code/njunmt_dist/configs/uy2zh/final/sample2_big_relative.yaml"
#export CON="configs/uy2zh/transformer_uy2zh_sample_big.yaml"
export SAVETODIR="/home/user_data55/wangdq/data/ccmt/uy-zh/output/"
mkdir -p $SAVETODIR


python -m src.bin.translate \
    --model_name "transformer" \
    --source_path "/home/user_data55/wangdq/data/ccmt/uy-zh/dev2020/dev.uy.token" \
    --model_path "/home/user_data55/liuzh/ccmt_model/uy2zh/sample_big_relative/transformer_relative.best.final"  \
    --ref_path "/home/user_data55/wangdq/data/ccmt/uy-zh/dev2020/dev.zh"  \
    --config_path "${CON}" \
    --batch_size 20 \
    --beam_size 5 \
    --alpha 1.5 \
    --use_gpu  \
    --saveto "${SAVETODIR}/sample1.out" \
    --num_refs 1  \
#    --sample_search_k 2
