pytorch 学习记录
摘要:
2023.10.31 model.train():启用 Batch Normalization 和 Dropout。作用:对BN层,保证BN层能够用到每一批数据的均值和方差,并进行计算更新;对于Dropout,model.train()是随机取一部分网络连接来训练更新参数。 model.eval() 阅读全文
posted @ 2023-10-31 11:48 Mju_halcyon 阅读(12) 评论(0) 推荐(0) 编辑