![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
jack_ya_ng
北航在读硕士,研究方向image captioning、object detection、DRL
展开
-
pytorch 迁移学习
迁移学习我们经常会用到,无论是CV还是NLP任务。在pytorch中主要通过如下方式进行:方式一:1)首先冻结不需要进行梯度更新曾的参数(减少计算):class Net(nn.Module): def __init__(self, model, pretrained): super(self, Net).__init__() self.resnet = model(pretrai...原创 2020-03-08 19:35:21 · 319 阅读 · 0 评论 -
pytorch DataParallel理解及易犯错误(逻辑上感觉没问题,但是返回时候却出错)
本文只针对单机多GPU使用dataparallel进行加速运算。写在前边: dataparallel只存在于继承了nn.Modules类的forward()计算中。大致流程如下:import torchmodel = Net() #初始化模型for i, (input_datas, label_datas) in enumerate(data_loader): #step 1: 数据...原创 2020-02-26 23:22:57 · 6928 阅读 · 17 评论