41 lines
1.6 KiB
Bash
41 lines
1.6 KiB
Bash
|
echo $$
|
||
|
|
CUDA_DEVICE=$1
|
||
|
|
DMOVE=$2
|
||
|
|
DLG=$3
|
||
|
|
|
||
|
|
echo $$ $CUDA_DEVICE $DMOVE $DLG
|
||
|
|
|
||
|
|
FOLDER="models/baselines_oracle"
|
||
|
|
mkdir -p $FOLDER
|
||
|
|
for SEED in 123; do #1 2 3 4 5; do #0; do #
|
||
|
|
for MODEL in LSTM; do # LSTM; do # Transformer; do #
|
||
|
|
for EXP in 2; do # 2 3; do
|
||
|
|
# for DMOVE in "No" "Yes"; do
|
||
|
|
# for DLG in "No" "Yes"; do
|
||
|
|
for INT in 0 1 2 3 4 5 6 7; do
|
||
|
|
FILE_NAME="plan_exp${EXP}_${MODEL}_dlg_${DLG}_move_${DMOVE}_int${INT}_seed_${SEED}"
|
||
|
|
COMM="plan_predictor_oracle.py"
|
||
|
|
COMM=$COMM" --seed=${SEED}"
|
||
|
|
COMM=$COMM" --device=${CUDA_DEVICE}"
|
||
|
|
COMM=$COMM" --use_dialogue_moves=${DMOVE}"
|
||
|
|
COMM=$COMM" --use_dialogue=${DLG}"
|
||
|
|
COMM=$COMM" --experiment=${EXP}"
|
||
|
|
COMM=$COMM" --seq_model=${MODEL}"
|
||
|
|
COMM=$COMM" --pov=First"
|
||
|
|
COMM=$COMM" --plan=Yes"
|
||
|
|
COMM=$COMM" --intermediate=${INT}"
|
||
|
|
if [ $INT -gt 0 ]; then
|
||
|
|
COMM=$COMM" --model_path=${FOLDER}/plan_exp${EXP}_${MODEL}_dlg_${DLG}_move_${DMOVE}_int0_seed_${SEED}.torch"
|
||
|
|
fi
|
||
|
|
COMM=$COMM" --save_path=${FOLDER}/${FILE_NAME}.torch"
|
||
|
|
echo $(date +%F\ %T)" python3 ${COMM} > ${FOLDER}/${FILE_NAME}.log"
|
||
|
|
nice -n 5 python3 $COMM > ${FOLDER}/${FILE_NAME}.log
|
||
|
|
done
|
||
|
|
# done
|
||
|
|
# done
|
||
|
|
done
|
||
|
|
done
|
||
|
|
done
|
||
|
|
|
||
|
|
echo 'Done!'
|