fcn epo 5

cafc769c · Aston Zhang · 9244de0c · cafc769c
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

chapter_computer-vision/fcn.md chapter_computer-vision/fcn.md +1 -1

未找到文件。
--- 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)
 ```

 ## 预测