
机器学习
文章平均质量分 53
dis_to_free
这个作者很懒,什么都没留下…
展开
-
训练神经网络太慢 DataLoader设置num_workers=4报错 RuntimeError: Caught RuntimeError in DataLoader worker process
在服务器上跑网络感觉很慢,尤其是几块显卡都有人在跑的时候,尝试在train_loader=DataLoader(train_data,batch_size=trainargs.batch_size,shuffle=True,num_workers=4) 设置num_workers=4,发现报错RuntimeError: Caught RuntimeError in DataLoader worker process。于是网上搜答案,但却没找到能成功解决的。于是放弃,就让它这么慢吧。直到有一天,心有不..原创 2021-07-04 10:37:38 · 10709 阅读 · 0 评论 -
pytorch 集合set里存放tensor变量无法求交集
import torcha=torch.tensor([1,2])b=torch.tensor([2,3])a=set(a)b=set(b)print(a&b)上述方式打印出为空集{},需要将tensor变量变成numpy变量import torchimport numpy as npa=np.array([1,2])b=np.array([2,3])a=set(a)b=set(b)print(a&b)如果原本是tensor变量,则通过.numpy()将.原创 2021-05-13 20:19:54 · 2954 阅读 · 0 评论 -
Tensorboard 报错 graph = graph_profile[0] TypeError: ‘GraphDef‘ object does not support indexing
想要用tensorboard显示网络计算图from tensorboardX import SummaryWriter ...... resultdict = G(input) if(count==0): with SummaryWriter(comment='graph') as w: w.add_graph(G,(input,))......报错warnings.warn(原创 2021-04-24 23:08:23 · 2002 阅读 · 0 评论 -
Ubuntu16.04(Ubuntu18) 安装 densepose
从开始安装到最后成功在服务器部署好,前后历经三个礼拜,虽说不是一直在装,但和一些大佬花几个晚上弄好差距也有点大,不过整个过程学到很多东西。下面大概讲一下整个过程遇到的一些印象深刻、耗时很久才解决的bug。如果大家不想看后面的碎碎念,那么直接去看这篇教程就好了,Installing DensePose,辅助参考PART1部分 全过程 和 吐血之路两篇教程。可以避开后面的很多坑。不过可能还是会遇到...原创 2019-10-27 23:32:36 · 1546 阅读 · 2 评论