pytorch学习笔记（十六）：pytorch 写代码时应该注意

1. 当网络中有 dropout，bn 的时候。训练的要记得 net.train(), 测试 要记得 net.eval()
2. 在测试的时候 创建输入 Variable 的时候 要记得 volatile=True
3. torch.sum(Tensor), torch.mean(Tensor) 返回的是 python 浮点数，不是 Tensor。
4. 在不需要 bp 的地方用 Tensor 运算。

我的博客即将同步至腾讯云+社区，邀请大家一同入驻。