From 3ab05ba0b3cf1b1cc7445f78765bb9d41f4d013f Mon Sep 17 00:00:00 2001 From: Nikita Araslanov Date: Mon, 2 Dec 2019 14:51:42 +0100 Subject: [PATCH] fixing #3 for main_preproc --- code/datasets/cvppp.lua | 2 -- code/runs/cvppp_preproc.sh | 17 ++++++++++++++--- 2 files changed, 14 insertions(+), 5 deletions(-) diff --git a/code/datasets/cvppp.lua b/code/datasets/cvppp.lua index 1e6be36..4359ef3 100644 --- a/code/datasets/cvppp.lua +++ b/code/datasets/cvppp.lua @@ -57,8 +57,6 @@ function CVPPPDataset:get(i) print('Warning: ' .. fullPath .. ' doesn\'t exists!') end msk = msk[1] - else - msk = msk[1] end return { diff --git a/code/runs/cvppp_preproc.sh b/code/runs/cvppp_preproc.sh index da6ef4a..f34f059 100644 --- a/code/runs/cvppp_preproc.sh +++ b/code/runs/cvppp_preproc.sh @@ -1,8 +1,9 @@ #!/bin/bash -CFG=611_preproc +# name of the experiment +# (will create a directory with checkpoints and logs) +CFG=001_preproc DIR=checkpoints/cvppp_${CFG} -SEED=0 source runs/utils.bash @@ -12,9 +13,19 @@ mkdir $DIR TRAIN_OPT="-learning_rate 1e-3 -batch_size 4 -iter_size 2 -learning_rate_decay 2e-5 -weightDecay 1e-4 -nEpochs 300" LOSS_OPT="-predictAngles -predictFg -seq_length 21 -preproc_mode" CMD="th main_preproc.lua -dataset cvppp -model_id $CFG $TRAIN_OPT $LOSS_OPT" -echo $CMD + +echo "Executing command: ${CMD}" + +# saving the changes +# to reproduce the run +echo `git diff` > $DIR/001_train.diff +echo `git rev-parse HEAD` > $DIR/001_train.head LOG=$DIR/001_train.log +echo "Logging: ${LOG}" nohup $CMD > $LOG 2>&1 & +sleep 1 +tail -f $LOG + exit 0;