PyTorch_YOLOv1 icon indicating copy to clipboard operation
PyTorch_YOLOv1 copied to clipboard

您好,我在训练您的程序时出现了一些问题

Open Student-Wang-wenjie opened this issue 2 years ago • 2 comments

报错 [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 avatar Dec 03 '22 12:12 Student-Wang-wenjie

@Student-Wang-wenjie 这些只是在测试flops和模型参数量时的warning信息,不是error,没有影响的~

yjh0410 avatar Dec 04 '22 00:12 yjh0410

@Student-Wang-wenjie 这些只是在测试flops和模型参数量时的warning信息,不是error,没有影响的~

好的,我已经了解,谢谢您的回复~

Student-Wang-wenjie avatar Dec 06 '22 07:12 Student-Wang-wenjie