Pytorch中的 model.train() 和 model.eval() 模式
摘要:
model.train() :启用 BatchNormalization 和 Dropout model.eval() :不启用 BatchNormalization 和 Dropout 把模型model.eval()作用是为了固定BN和dropout层,使得偏置参数不随着发生变化。因为当batch 阅读全文
posted @ 2020-08-21 22:58 cltt 阅读(390) 评论(0) 推荐(0) 编辑