240 发简信
IP属地:西藏
  • 博主,想请教一下您,我现在只想训练linear层中两个w,W(w1,w2,w3,......w100),现在只想训练w1,w2,该怎样去选择这两个权重去训练呢?

    pytorch固定部分参数进行网络训练

    1. 只训练部分层 这样就将for循环以上的参数固定, 只训练下面的参数。但是注意需要在optimizer中添加上这样的一句话filter(lambda p: p.requi...