BUG调试
文章平均质量分 92
清新大叔丶
我的忍道就是决不放弃!
展开
-
pytorch分布式训练踩坑记录
pytorch分布式训练踩坑记录近期工作涉及到Backbone,所以会涉及到ImageNet1K(120w+张图像)的预训练先说说手上有啥,一台PC,配置如下2 张 2080ti500G SSD 2T 硬盘64 内存cpu i7 8核任务是图像分类,虽然简单但是大型数据库的训练极其耗时。 无奈实验室设备就那么多(留下没多张V100的泪水)硬着头皮也得上git 上的Pipeline 其实也不少,这里我推荐俩https://github.com/JaminFong/dali-pytorch原创 2021-11-21 22:18:24 · 4549 阅读 · 1 评论 -
如何解决 ValueError: operands could not be broadcast together with shapes
在复现DHSnet中碰到了这个问题:使用的数据集为ECSSD出错的代码行为该函数的作用为将输入进来的图片和真值进行去均值处理,以增强物体与背景的对比度,最后转化为GPU可以处理的张量。而为什么会在这一行出错呢,直译过来就是操作数不能与形状一起广播,关于np的广播机制,这篇博文有详细的记载我就不赘述了。这么一看,就推测img中的向量纬度出错导致不能相减。将代码改为并运行结果发现打印...原创 2019-08-02 20:29:28 · 34491 阅读 · 3 评论