Skip to content

Commit

Permalink
fixing #3 for main_preproc
Browse files Browse the repository at this point in the history
  • Loading branch information
arnike committed Dec 2, 2019
1 parent 0433f12 commit 3ab05ba
Show file tree
Hide file tree
Showing 2 changed files with 14 additions and 5 deletions.
2 changes: 0 additions & 2 deletions code/datasets/cvppp.lua
Original file line number Diff line number Diff line change
Expand Up @@ -57,8 +57,6 @@ function CVPPPDataset:get(i)
print('Warning: ' .. fullPath .. ' doesn\'t exists!')
end
msk = msk[1]
else
msk = msk[1]
end

return {
Expand Down
17 changes: 14 additions & 3 deletions code/runs/cvppp_preproc.sh
Original file line number Diff line number Diff line change
@@ -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

Expand All @@ -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;

0 comments on commit 3ab05ba

Please sign in to comment.