Skip to content

Commit 5f501da

Browse files
author
Matt Crane
committed
Forgot cudnn from model/logs
1 parent 10a714f commit 5f501da

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

run.sh

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -107,9 +107,9 @@ if [[ $1 = "all" || $1 = "gpu" ]]; then
107107
do
108108
for dataset in "TrecQA" "WikiQA"
109109
do
110-
${GPU_DOCKER} run ${GPU_RUNTIME} -it --name qqa snapbug/qqa:sha-cf0e269 sh -c "python main.py qqa.${dataset}.model.gpu --paper-ext-feats --cuda ${cudnn} --dataset_folder=../../data/${dataset}"
111-
${GPU_DOCKER} logs ${GPU_RUNTIME} qqa > qqa.${dataset}.log.gpu
112-
${GPU_DOCKER} cp ${GPU_RUNTIME} qqa:/castorini/castor/sm_cnn/qqa.${dataset}.model.gpu .
110+
${GPU_DOCKER} run ${GPU_RUNTIME} -it --name qqa snapbug/qqa:sha-cf0e269 sh -c "python main.py qqa.${dataset}.model.gpu${cudnn} --paper-ext-feats --cuda ${cudnn} --dataset_folder=../../data/${dataset}"
111+
${GPU_DOCKER} logs ${GPU_RUNTIME} qqa > qqa.${dataset}.log.gpu${cudnn}
112+
${GPU_DOCKER} cp ${GPU_RUNTIME} qqa:/castorini/castor/sm_cnn/qqa.${dataset}.model.gpu${cudnn} .
113113
${GPU_DOCKER} rm ${GPU_RUNTIME} qqa
114114
done
115115
done

0 commit comments

Comments
 (0)