pytorch载入预训练模型后,实现训练指定层
在传入optimizer时，和一般的传参方法torch.optim.Adam 不同，参数部分用了一个list， list的每个元素有params和lr两个键值。如果没有 lr则应用Adam的lr属性。Adam的属性除了lr， 其他都是参数所共有的。
下载地址
用户评论
