![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
代码修改
文章平均质量分 67
weixin_42160956
这个作者很懒,什么都没留下…
展开
-
数据并行加载在多个gpu上
一般数据读取会涉及两个函数:1.get_dataset(data_path,cfg)。其中的__getitem__(self,idx)函数,这个函数用于读取给定路径下的数据并处理成所需的格式,返回一个数据的data,data一般为字典格式。作用:规范数据,path-->tensor(data)。2.dataloader(get_dataset),可直接调用pytorch中的接口。通过for i ,data_batch in enumerate(dataloader)调用前文提到的_.原创 2020-08-07 14:14:16 · 2133 阅读 · 0 评论 -
深度学习网络中设置部分层参数可调
1.基于框架pytorch。先查看网络每层的名字p[0],将需要梯度更新的层的名字记下,替换以下代码中的’conv5_ft.conv.weight/bias’,将不需要梯度更新层的require_grad设为False。for p in net.named_parameters():if p[0] != ‘conv5_ft.conv.weight’ and p[0]!=‘conv5_ft.co......原创 2019-05-29 09:34:51 · 766 阅读 · 0 评论