- 当网络中有
dropout,bn
的时候。训练的要记得net.train()
, 测试 要记得net.eval()
- 在测试的时候 创建输入
Variable
的时候 要记得volatile=True
torch.sum(Tensor), torch.mean(Tensor)
返回的是python
浮点数,不是Tensor
。- 在不需要
bp
的地方用Tensor
运算。
pytorch学习笔记(十六):pytorch 写代码时应该注意
最新推荐文章于 2022-11-12 20:21:19 发布