pytorch
再困也得吃
这个作者很懒,什么都没留下…
展开
-
pytorch——AttributeError: 'DataParallel' object has no attribute '****'
报错原因:在使用model = nn.DataParallel(model,device_ids=[0,1])加载模型之后,出现了这个错误:AttributeError: ‘DataParallel’ object has no attribute ‘****’报错的地方在我后面调用model的一些层时,并没有那些层,输出经过nn.DataParallel的模型参数后,发现每个参数前面多了m...原创 2020-03-26 17:02:29 · 13215 阅读 · 1 评论 -
pytorch——多GPU训练,CPU测试出现的问题
我在训练模型的时候使用了nn.DataParallel,并在在服务器gpu测试也同时将训练好的模型并行载入,这样没有问题,但是我将训练好的模型在本地cpu运行的时候出错。这里记录出错的地方,并记录大致流程。1. CPU读取模型:RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_av...原创 2020-03-06 20:52:00 · 1909 阅读 · 0 评论 -
torchvision 的 transforms 与 python 的图像接口opencv、skimage和PIL 相关总结
torchvision的transforms里面包含了很多数据增强的方法,但是网上看到很多人采用了很多不同的图像接口读取图片,而出现很多错误,这里总结一下python的图像接口格式的区别已经转化方法 以及 transforms的使用。1.python 图像接口基本操作1.1 opencv读取图片1.1.1 读取方法# 默认彩图img_cv2 = cv2.imread('test.jpg'...原创 2019-05-17 17:31:16 · 4371 阅读 · 1 评论 -
pytorch tensor和ndarray转换相关总结
在使用pytorch的时候,经常会涉及到两种数据格式tensor和ndarray之间的转换,这里总结一下两种格式的转换:1. tensor cpu 和tensor gpu之间的转化:tensor cpu 转为tensor gpu:tensor_gpu = tensor_cpu.cuda()>>> tensor_cpu = torch.ones((2,2))tenso...原创 2020-01-06 10:23:00 · 5770 阅读 · 1 评论 -
pytorch 本地远程使用 visdom
pytorch 远程使用 visdom安装 visdompip install visdom启动python -m visdom.server若出现OSError: [Errno 98] Address already in use visdom,即说明端口被占用。解决方法方法一查看端口占用情况(vidsom 默认占用8097端口)lsof -i:8097中断...原创 2019-09-28 23:07:29 · 983 阅读 · 0 评论 -
pytorch 指定显卡
在pytorch中若使用gpu, 则默认从0号卡开始终端中指定CUDA_VISIBLE_DEVICES=1 python mian.py代码中指定import osos.environ["CUDA_VISIBLE_DEVICES"] = "0"原创 2020-01-11 21:05:49 · 1921 阅读 · 0 评论