Pruning 段错误 about mobile-yolov5-pruning-distillation

[dengly]

python train.py --type mvocs 的 epochs是300
python train.py --type smvocs 的 epochs是100

[Syencil]

这描述给的信息太少了 尝试单步debug吧 我这边流程是没问题

[dengly]

这描述给的信息太少了 尝试单步debug吧 我这边流程是没问题

确认一下我这样的操作是否正常

[dengly]

这个是我执行的脚步

# 训练 Baseline
python train.py --type mvocs

# 稀疏 sparse learning
python3 train.py --type smvocs

# 剪枝 Pruning
python3 pruning.py -t 0.1

# 微调 finetune
python3 train.py --type fsmvocs

# 蒸馏 Distillation
python3 train.py --type dmvocs

[dengly]

我修改了 cfg、data、weights、epochs、batch_size

[dengly]

baseline用的模型是 mobilenet_v2-b0353104.pth，后续的都是上一次训练得到的best模型，到pruning这步就这样了。我尝试重新训练看看是否正常

[TongShuo-Demon]

这个是我执行的脚步

# 训练 Baseline
python train.py --type mvocs

# 稀疏 sparse learning
python3 train.py --type smvocs

# 剪枝 Pruning
python3 pruning.py -t 0.1

# 微调 finetune
python3 train.py --type fsmvocs

# 蒸馏 Distillation
python3 train.py --type dmvocs

能问问您关于这个数据集放置的问题吗

[Syencil]

走debug吧 看是哪个位置错了

[dengly]

定位的执行bn_analyze方法的plt.hist(bn_val, bins=101, align="mid", log=True, range=(0, 1.0))报错

[Syencil]

如果是服务器terminal应该不带显示功能吧 plt的问题

[dengly]

定位到是matplotlib.pyplot相关方法调用报错，我注掉后使用IDEA运行正常，但是通过终端执行还是会报段错误。

[dengly]

如果是服务器terminal应该不带显示功能吧 plt的问题

我用的是深度的系统，deepin v20

[TongShuo-Demon]

test.py文件使用剪枝后的模型会报错，报错信息如下：
Traceback (most recent call last):
File "/home/demon/PycharmProjects/mobile-yolov5-pruning-distillation/test.py", line 357, in
test(opt.data,
File "/home/demon/PycharmProjects/mobile-yolov5-pruning-distillation/test.py", line 89, in test
names = {k: v for k, v in enumerate(model.names if hasattr(model, 'names') else model.module.names)}
File "/home/demon/anaconda3/envs/pytorch_gpu/lib/python3.8/site-packages/torch/nn/modules/module.py", line 771, in getattr
raise ModuleAttributeError("'{}' object has no attribute '{}'".format(
torch.nn.modules.module.ModuleAttributeError: 'Model' object has no attribute 'module'