随笔分类 - PyTorch

【PyTorch】按照 steps 训练和保存模型

摘要：当数据集很小时，训练多久用 epoch 表示不太合适，这个时候使用模型更新次数来表示更加合理，每多少个 steps 查看一次验证集效果并保存模型。阅读全文

posted @ 2020-09-30 16:59 wuliytTaotao 阅读(1981) 评论(0) 推荐(0) 编辑

【PyTorch】L2 正则化

摘要：PyTorch，只对卷积层和全连接层的 weights 进行 L2（weight decay）。阅读全文

posted @ 2020-06-25 20:35 wuliytTaotao 阅读(4087) 评论(0) 推荐(0) 编辑