fcn epo 5

wagnlinzh · Jul 20, 2018 · cafc769 · cafc769
1 parent 9244de0
commit cafc769
Showing 1 changed file with 1 addition and 1 deletion.
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=2)
+gb.train(train_iter, test_iter, net, loss, trainer, ctx, num_epochs=5)
 ```
 
 ## 预测