ultralytics / xview-yolov3

xView 2018 Object Detection Challenge: YOLOv3 Training and Inference.
https://docs.ultralytics.com
GNU Affero General Public License v3.0
256 stars 57 forks source link

Unable to train #18

Closed starofgit closed 5 years ago

starofgit commented 5 years ago

I am getting the following error while training: Traceback (most recent call last): File "train.py", line 209, in main(opt) File "train.py", line 131, in main weight=class_weights, epoch=epoch).to(device) File "/home/xxxxx/anaconda3/lib/python3.5/site-packages/torch/nn/modules/module.py", line 489, in call result = self.forward(*input, **kwargs) File "/home/xxxxx//xview/xview-yolov3-master/models.py", line 234, in forward x, losses = module[0](x, targets, requestPrecision, weight, epoch) ValueError: too many values to unpack (expected 2)

I have checked that above call is returning 14 values. Is it because of torch version mismatch?

glenn-jocher commented 5 years ago

Hello, thank you for your interest in our work! This is an automated response. Please note that most technical problems are due to:

If none of these apply to you, we suggest you close this issue and raise a new one using the Bug Report template, providing screenshots and minimum viable code to reproduce your issue. Thank you!