2021年11月10日
model.train()与model.eval()的用法
摘要： model.train()与model.eval()的用法 在深度学习的训练和测试代码中，总会有model.train()和model.eval()这两句，那么这两条语句的作用是什么？ 通过查阅发现： 如果模型中有BN层(Batch Normalization)和Dropout，需要在训练时添加mo 阅读全文
posted @ 2021-11-10 10:54 重大的小鸿 阅读(3555) 评论(0) 推荐(0) 编辑