![](https://img-blog.csdnimg.cn/20201014180756925.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
文章平均质量分 56
_Vixerunt
这个作者很懒,什么都没留下…
展开
-
【pytorch系列】多GPU并行训练 torch.nn.DataParallel用法
【问题】在使用nn.DataParallel时出现“ optimizer got an empty parameter list”问题。model = nn.DataParallel(model, device_ids=[0,1,2,3])optimizer = torch.optim.Adam(model.parameters(), lr=0.001, betas=(0.9, 0.999), eps=1e-08, weight_decay=0, amsgrad=False)[ValueError]原创 2022-03-02 11:12:20 · 3618 阅读 · 0 评论 -
【pytorch】nn.add_module(name,Module),children()
nn.add_module(name,Module)将一个 child module 添加到当前 modle。 被添加的module可以通过 name属性来获取。 例:import torch.nn as nnclass MyModule(nn.module): def __init__(self): super(MyModule,self).__init__() self.add_module('conv', nn.Conv2d(10, 20, 4)) #self.conv = nn.原创 2022-03-07 14:00:42 · 1473 阅读 · 0 评论