BUG记录
菜鸡是谁
这个作者很懒,什么都没留下…
展开
-
分层设置不同的学习率
分层设置不同的学习率对于自己定义的网络层(插入、加入…)等均可以设置不同的学习率,具体实现如下:对于某个单层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 阅读 · 0 评论