hi, I tried to train a 2-class model. i used your pretrained model .but error likes this:
Check failed: from.shape() == to->shape() operands shape mismatchfrom.shape = (125,) to.shape=(30,)
my command is : python train.py --train-path /data/data/train.rec --val-path /data/data/val.rec --pretrained /data/model/yolo2_darknet19_416 --epoch 0
ps ,i have changed the class_num to 1.