PyTorch_YOLOv1
PyTorch_YOLOv1 copied to clipboard
您好,我在训练您的程序时出现了一些问题
报错 [WARN] Cannot find rule for <class 'backbone.resnet.BasicBlock'>. Treat it as zero Macs and zero Params. [WARN] Cannot find rule for <class 'torch.nn.modules.container.Sequential'>. Treat it as zero Macs and zero Params. [WARN] Cannot find rule for <class 'backbone.resnet.ResNet'>. Treat it as zero Macs and zero Params. [WARN] Cannot find rule for <class 'models.basic.SPP'>. Treat it as zero Macs and zero Params.
运行代码如下:python train.py --cuda -d voc --batch_size 32 --lr 0.001 --max_epoch 150
能帮忙看下是什么原因吗? 谢谢您!
@Student-Wang-wenjie 这些只是在测试flops和模型参数量时的warning信息,不是error,没有影响的~
@Student-Wang-wenjie 这些只是在测试flops和模型参数量时的warning信息,不是error,没有影响的~
好的,我已经了解,谢谢您的回复~