pytorch
想进步的小孟
这个作者很懒,什么都没留下…
展开
-
自动学习率调整
参考 MultiStepLR参数 MultiStepLR(optimizer, milestones=[20,60,80], gamma=0.1) optimizer:自己设定的优化器 milestones:到达这个epoch时学习率变化,是一个列表 gamma:学习率变化的倍率,到达第20个epoch时学习率由lr变为lr*0.1 代码 import torch.optim as optim from torch.optim.lr_scheduler import MultiStepLR from mat原创 2021-12-22 11:17:56 · 549 阅读 · 0 评论 -
卷积神经网络模型参数计算
参考文章 神经网络模型可视化工具、参数量、Flops 统计工具 使用torchsummary打印torch每层形状 AttributeError: ‘list’ object has no attribute 'size’解决方案 方式一:使用torchsummary import torch from torchsummary import summary from pytorch_model_attention import densenet121 device = torch.device('cud原创 2021-12-21 16:57:09 · 395 阅读 · 0 评论