提交 39cc5c9b 编写于 作者: A Aston Zhang

fcn epo 3

上级 121a6a39
......@@ -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=5)
gb.train(train_iter, test_iter, net, loss, trainer, ctx, num_epochs=3)
```
## 预测
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册