Skip to content

Commit a2aa13e

Browse files
authored
add save_interval to tutorial training command line
1 parent 64eb1ac commit a2aa13e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tutorial/train_shape.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -305,7 +305,7 @@
305305
},
306306
"source": [
307307
"# consider this is a simple dataset, train head will be enough.\n",
308-
"! python train.py -c 0 -p shape --head_only True --lr 1e-3 --batch_size 32 --load_weights weights/efficientdet-d0.pth --num_epochs 50\n",
308+
"! python train.py -c 0 -p shape --head_only True --lr 1e-3 --batch_size 32 --load_weights weights/efficientdet-d0.pth --num_epochs 50 --save_interval 100\n",
309309
"\n",
310310
"# the loss will be high at first\n",
311311
"# don't panic, be patient,\n",

0 commit comments

Comments
 (0)