search

Bohao-Lee / CME

64 stars 16 forks source link

RuntimeError: shape ‘[30, 5, 6, 13, 13]’ is invalid for input of size 5070 #33

Open Zhengzkang opened 2 years ago

Zhengzkang commented 2 years ago

Hi，Lee，I want to train VOC2007 and 2012 in which the 20 classes are set to base classes, and my own dataset's 7 classes are set to novel classes, But, I get a trouble after one epoch training, can you give me some suggestions? By the way, I run "python train_meta.py cfg/metayolo.data cfg/darknet_dynamic.cfg cfg/reweighting_net.cfg darknet19_448.conv.23 "

Zhengzkang commented 2 years ago

image

Zhengzkang commented 2 years ago

如果正常训练VOC的话，可以正常训练，但是加入自己的数据集和类别后，训练1轮或多轮后就会出现这种报错

shizhongwu commented 2 years ago

如果正常训练VOC的话，可以正常训练，但是加入自己的数据集和类别后，训练1轮或多轮后就会出现这种报错

你好，我在使用作者给定的数据集训练时也遇到了这种情况，我想问一下你这个问题解决了吗？