search

talebolano / yolov3-network-slimming

yolov3 network slimming剪枝的一种实现
345 stars 93 forks source link

x, losses = self.module_list[i][0](x, targets) #47

Open ZhaoMonica opened 4 years ago

ZhaoMonica commented 4 years ago

x, losses = self.module_list[i][0](x, targets) ValueError: too many values to unpack (expected 2)

How to solve this problem？

liwenjia41 commented 4 years ago

x, losses = self.module_list[i][0](x, targets) ValueError: too many values to unpack (expected 2)

How to solve this problem？

change losses --> losses DetectionLayer return type is tuple: (loss, loss_x.item(), loss_y.item(), loss_w.item(), loss_h.item(), loss_conf.item(), loss_cls.item(), recall, precision,) So x = loss and losses = (other parameters)