mmpretrain icon indicating copy to clipboard operation
mmpretrain copied to clipboard

VOC2007 Train, RuntimeError: Expected floating point type for target with class probabilities, got Char

Open Recialhot opened this issue 2 years ago • 1 comments

when I use the resnet50 to train VOC2007 datasets, it will appear image

voc_bs16.py: image

base = [ '../base/models/resnet50.py', '../base/datasets/voc_bs16.py', '../base/schedules/cifar10_bs128.py', '../base/default_runtime.py' ]

Recialhot avatar Jun 22 '22 14:06 Recialhot

VOC is a multi_label dataset, if you want to use it, you should use the multi_label head.

Ezra-Yu avatar Jun 23 '22 02:06 Ezra-Yu

Now, we have provided the mutil_label task config, please refer to csra

Ezra-Yu avatar Sep 08 '22 02:09 Ezra-Yu