Deep learning
PhDing_H
这个作者很懒,什么都没留下…
展开
-
pytorch-mmsegmentation train时遇到AssertionError:Default process group is not initialized
问题原因:非分布式训练使用了分布式训练的设置 两种解决办法: 1、在tools/train.py 中加入 import torch.distributed as dist dist.init_process_group('gloo', init_method='file:///temp/somefile', rank=0, world_size=1) 2、在configs/_base_/models 的文件中,首行norm_cfg = dict(type='SyncBN', requires_gr原创 2020-11-18 20:35:42 · 8786 阅读 · 12 评论 -
Error when checking target: expected 'layers' to have 4 dimensions, but got array with shape (num,2)
这种问题出现在classification model 中可能是模型的最后一层池化层后没有加flatten层,导致上述错误原创 2019-12-13 10:20:13 · 646 阅读 · 2 评论