fcn ep 2, build doc rm voc

wagnlinzh · Jul 20, 2018 · 9244de0 · 9244de0
1 parent 39cc5c9
commit 9244de0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/build/build_doc.sh b/build/build_doc.sh
@@ -3,6 +3,8 @@
 set -x
 set -e
 
+rm -rf data/VOCdevkit
+
 # Clean build/chapter*/*ipynb and build/chapter*/*md that are no longer needed.
 cd build
 for ch in chapter*; do

diff --git a/chapter_computer-vision/fcn.md b/chapter_computer-vision/fcn.md
@@ -146,7 +146,7 @@ train_iter = gdata.DataLoader(gb.VOCSegDataset(True, input_shape, voc_dir),
 test_iter = gdata.DataLoader(gb.VOCSegDataset(False, input_shape, voc_dir),
                              batch_size, last_batch='discard', num_workers=4) 
 
-gb.train(train_iter, test_iter, net, loss, trainer, ctx, num_epochs=3)
+gb.train(train_iter, test_iter, net, loss, trainer, ctx, num_epochs=2)
 ```
 
 ## 预测