TypeError: cannot unpack non-iterable NoneType object

[7W7W7W]

您好，感谢您的工作，我在改进后的模型中使用您的方法后出现以下错误，请问这个怎么处理呢？谢谢！

Traceback (most recent call last): File "f:/yolov5-7.0/compress.py", line 612, in main(opt) File "f:/yolov5-7.0/compress.py", line 592, in main train(opt.hyp, opt, device, callbacks) File "f:/yolov5-7.0/compress.py", line 251, in train pruned_model = compress(model, val_loader if RANK in [-1, 0] else None, opt) File "f:/yolov5-7.0/compress.py", line 88, in compress sen_dict = sens(model, dataloader, args.part) File "f:\yolov5-7.0\prune\sensitivity.py", line 348, in call _, _, base_flops = self.model.cuda().info(False, self.inputsize) TypeError: cannot unpack non-iterable NoneType object

[HankYe]

Hi, thanks for ur attention to our work. It seems that self.model.cuda().info(False, self.inputsize) return a Nonetype object. But from model_info() function, it should return a tuple. So I suggest you double-check the variable type of self.model.