Skip to content

YoloV3 from lightnet 1.1.1 : wrong data type in _regionloss.py

Trying to use YoloV3 from lightnet 1.1.1 on my custom dataset, the package threw me the following error :

Exception has occurred: AttributeError
'list' object has no attribute 'data'
  File "C:\some\path\lightnet\network\loss\_regionloss.py", line 97, in forward
    nB = output.data.size(0)

indeed output is a list. Isn't it because somewhere output[0] should be performed ?

EDIT

looking further, I see that at the beginning of the call i.e. in process_batch of the training engine, out is a list that I assume should get flatten by line 489 of module.py from PyTorch (base class that I suppose defines the loss method) :

result = self.forward(*input, **kwargs)

but somehow the input (that I assume should be a list) got encapsulated in a tuple and this may be what causes the error ?

Edited by FenryrMKIII