- 博客(2)
- 收藏
- 关注
转载 分层设置不同的学习率
分层设置不同的学习率对于自己定义的网络层(插入、加入…)等均可以设置不同的学习率,具体实现如下:对于某个单层net = net()lr = 0.001 conv5_params = list(map(id, net.conv5.parameters()))base_params = filter(lambda p: id(p) not in conv5_params, ...
2020-02-09 10:56:23 1062 1
原创 pytorch中多GPU并行报错
pytorch中多GPU并行报错RuntimeError: Expected tensor for argument #1 ‘input’ to have the same device as tensor for argument #2 ‘weight’; but device 1 does not equal 0 (while checking arguments for cudnn_conv...
2020-02-08 22:00:35 1257
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人