1. 首页
  2. 编程语言
  3. Python
  4. pytorch载入预训练模型后,实现训练指定层

pytorch载入预训练模型后,实现训练指定层

上传者: 2022-03-15 21:56:01上传 PDF文件 55.94 KB 热度 18次

在传入optimizer时,和一般的传参方法torch.optim.Adam 不同,参数部分用了一个list, list的每个元素有params和lr两个键值。如果没有 lr则应用Adam的lr属性。Adam的属性除了lr, 其他都是参数所共有的。

用户评论