search

sufeidechabei / gluon-mobilenet-yolov3

Gluon-Mobilenet-yolov3
98 stars 33 forks source link

这个训练的代码有问题，训练时会报错 #12

Open benz725 opened 4 years ago

benz725 commented 4 years ago

第307行的代码创建了一个LRScheduler的类，但是这个的init方法里面并没有step和warmup_epochs两个关键字，因此会报如下错误: Traceback (most recent call last): File "/home/dell/PycharmProjects/gluon-mobilenet-yolov3-master/train_yolo3_mobilenet.py", line 488, in train(net, train_data, val_data, eval_metric, ctx, args) File "/home/dell/PycharmProjects/gluon-mobilenet-yolov3-master/train_yolo3_mobilenet.py", line 313, in train warmup_epochs=args.warmup_epochs) TypeError: init() got an unexpected keyword argument 'step

benz725 commented 4 years ago

看到另外的人写的了，是gluoncv版本的问题，切换成0.3.0可能会解决