![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Ptorch学习笔记
璐璐yao
追梦璐
展开
-
AssertionError: Invalid device id
文章目录pytorch报错AssertionError: Invalid device idpytorch报错AssertionError: Invalid device id代码:model = torch.nn.DataParallel(model, device_ids=[5,7]).cuda()因为我想使用两个GPU,但是出现报错,原因是因为pytorch默认使用cuda=0的GPU,当gpu编号为device:0的设备被占用时,指定其他编号gpu使用torch.nn.DataParal原创 2020-06-04 18:32:03 · 12402 阅读 · 3 评论 -
HRnet网络结构分析(deep high-resolution representation learning for human pose estimation)
这几天认真的看了pose_hrnet.py的代码,源码地址:https://github.com/leoxiaobin/deep-high-resolution-net.pytorch总结HRnet网络结构图:原创 2020-04-03 21:47:31 · 833 阅读 · 0 评论 -
transfer learning
Transfer learning 迁移学习介绍因为拥有足够大的数据集相对很少,因此通常在很大的数据集上对ConvNet进行预训练,然后将ConvNet用作初始化或固定特征提取器以完成感兴趣的任务。三种常见的transfer learning的方法(1)ConvNet作为固定特征提取器删除最后一个完全连接层,然后将其余剩下的部分看作固定特征提取器(2)微调ConvNet通过在新的数据...翻译 2020-02-20 18:10:01 · 230 阅读 · 0 评论 -
Pytorch学习笔记
Pytorch 如何定义自己的数据集数据集: Penn-Fudan Database for Pedestrian Detection and Segmentation.(Penn-Fudan数据集)前言在学习pytorch官网教程时,作者对Penn-Fudan数据集进行了定义,并且在自定义的数据集上实现了对R-CNN模型的微调。此篇笔记简单总结一下pytorch如何实现定义自己的数据集...原创 2020-02-16 16:49:52 · 598 阅读 · 2 评论 -
pytorch学习笔记
在学习pytorchtutorial时,出现错误AttributeError: module ‘tensorflow_core._api.v2.io.gfile’ has no attribute 'get_filesystem’原因:tensorflow和tensorboard发生冲突解决方法:在原代码中加入以下代码:import tensorflow as tfimport ten...原创 2020-02-11 14:42:10 · 227 阅读 · 0 评论 -
Pytorch学习笔记
Pytorch FashionMnist网上数据集下载缓慢的解决方法今天用pytorch下载FashionMnist数据集的时候出现下载缓慢的情况,于是打算从本地导入数据集,过程总结如下:1.下载网址Fashionmnist本地下载Fashionmnist下载好文件以后格式为将下载好的文件放到特定的文件夹下,此时我用的路径为 E:\Code\PycharmProjects\Fahio...原创 2020-02-10 22:07:26 · 147 阅读 · 4 评论