2019年5月7日
Pytorch中的 model.train() 和 model.eval() 模式
摘要： model.train() ：启用 BatchNormalization 和 Dropout model.eval() ：不启用 BatchNormalization 和 Dropout 参考： https://pytorch.org/docs/stable/nn.html?highlight=mo 阅读全文
posted @ 2019-05-07 15:23 sbj123456789 阅读(37652) 评论(0) 推荐(2) 编辑