mmpretrain
mmpretrain copied to clipboard
VOC2007 Train, RuntimeError: Expected floating point type for target with class probabilities, got Char
when I use the resnet50 to train VOC2007 datasets, it will appear
voc_bs16.py:
base = [ '../base/models/resnet50.py', '../base/datasets/voc_bs16.py', '../base/schedules/cifar10_bs128.py', '../base/default_runtime.py' ]
VOC is a multi_label dataset, if you want to use it, you should use the multi_label head.
Now, we have provided the mutil_label task config, please refer to csra