2020年9月19日
Pytorch trick集锦
摘要： 1.查看模型每层输出详情 1 from torchsummary import summary 2 summary(your_model, input_size=(channels, H, W)) input_size是根据你自己的网络模型的输入尺寸进行设置。 2.梯度裁剪 1 import tor 阅读全文
posted @ 2020-09-19 14:57 最咸的鱼 阅读(343) 评论(0) 推荐(0) 编辑