SpursLipu / YOLOv3v4-ModelCompression-MultidatasetTraining-Multibackbone

YOLO ModelCompression MultidatasetTraining
GNU General Public License v3.0
444 stars 136 forks source link

报错TypeError #114

Closed Biubiubiu12 closed 3 years ago

Biubiubiu12 commented 3 years ago

Starting training for 10 epochs...

 Epoch   gpu_mem      GIoU       obj       cls     total   targets  img_size
   0/9     6.87G      5.61      3.01         0      8.62         5       416: 100%|███████████████████████████████████████| 1202/1202 [07:38<00:00,  2.62it/s]
           Class    Images   Targets         P         R   mAP@0.5        F1: 100%|████████████████████████████████████████▊| 205/206 [00:32<00:00,  6.26it/s]

Traceback (most recent call last): File "train.py", line 632, in train(hyp) # train normally File "train.py", line 489, in train FPGA=opt.FPGA) File "/home/cxzx/a-slr-work/YOLOv3v4-ModelCompression-MultidatasetTraining-Multibackbone-master/test.py", line 95, in test inf_out, trainout, = model(imgs, augment=augment) # inference and training outputs File "/home/cxzx/.local/lib/python3.6/site-packages/torch/nn/modules/module.py", line 727, in _call_impl result = self.forward(*input, kwargs) File "/home/cxzx/.local/lib/python3.6/site-packages/torch/nn/parallel/distributed.py", line 621, in forward outputs = self.parallel_apply(self._module_copies[:len(inputs)], inputs, kwargs) File "/home/cxzx/.local/lib/python3.6/site-packages/torch/nn/parallel/distributed.py", line 646, in parallel_apply return parallel_apply(replicas, inputs, kwargs, self.device_ids[:len(replicas)]) File "/home/cxzx/.local/lib/python3.6/site-packages/torch/nn/parallel/parallel_apply.py", line 86, in parallel_apply output.reraise() File "/home/cxzx/.local/lib/python3.6/site-packages/torch/_utils.py", line 428, in reraise raise self.exc_type(msg) TypeError: Caught TypeError in replica 1 on device 1. Original Traceback (most recent call last): File "/home/cxzx/.local/lib/python3.6/site-packages/torch/nn/parallel/parallel_apply.py", line 61, in _worker output = module(*input, *kwargs) File "/home/cxzx/.local/lib/python3.6/site-packages/torch/nn/modules/module.py", line 727, in _call_impl result = self.forward(input, kwargs) TypeError: forward() missing 1 required positional argument: 'x'

对数据集为hand进行训练,使用yolov3-608.weights作为预训练模型,在进行第2epoch时报错