pytorch学习笔记
pipihuaniupi
这个作者很懒,什么都没留下…
展开
-
OneCycleLR 训练中断恢复问题
pytorch 利用 torch.optim.lr_scheduler.OneCycleLR 更新学习率,中断训练之后,恢复训练。原创 2023-03-02 14:27:38 · 433 阅读 · 0 评论 -
pytorch 笔记 01 仅训练个别层,且设置不同学习率
本人学习交流使用参考自:https://blog.csdn.net/qq_19168521/article/details/117200115https://www.it1352.com/2111804.htmlimport torchimport torch.nn as nnclass test(nn.Module): def __init__(self): super(test, self).__init__() self.conv1 = nn原创 2021-12-20 10:37:05 · 426 阅读 · 0 评论