Skip to content

Commit

Permalink
add batch size option
Browse files Browse the repository at this point in the history
  • Loading branch information
jtkim-kaist committed Jan 12, 2018
1 parent c8bec1c commit bc54dc2
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion train.sh
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,6 @@ ckpt_update=`realpath $pylibdir/update_ckpt.py`

python3 $train -m 0 -e 1 --train_step=100 --batch_size=256 --prj_dir=$curdir

a
# ckpt_update script options
# u : update checkpoint from trained model (1) or restore checkpoint to default (0)
# Note that when u==0, the normalization factor is also restored to default.
Expand Down

0 comments on commit bc54dc2

Please sign in to comment.