2021年3月11日
关于Pytorch中model.train（）
摘要： model.train（）将模型设置为训练状态，作用：使Dropout，batchnorm知道后有不同表现（具体参考Dropout，batchnorm源码），只有这两个关心True or False。 将模型设置为测试状态有两种方法： 1.model.train(mode=False) 2.mode 阅读全文
posted @ 2021-03-11 19:53 吴莫愁258 阅读(2269) 评论(0) 推荐(0) 编辑