AttributeError: 'tuple' object has no attribute 'log_softmax' about volo HOT 2 OPEN

Do you use token label? If not, maybe you should modify "return_dense=True, mix_token=True" to "return_dense=False, mix_token=False" in

volo/models/volo.py

Line 462 in 068a583

post_layers=None, return_mean=False, return_dense=True, mix_token=True,

when i fixed your pointed codes, it appeared this error :
File "main.py", line 652, in main
train_metrics = train_one_epoch(epoch,
File "main.py", line 786, in train_one_epoch
loss = loss_fn(output, target)
File "C:\Program Files\Anaconda3\envs\yolov5-v4.0\lib\site-packages\torch\nn\modules\module.py", line 727, in _call_impl
result = self.forward(*input, **kwargs)
File "C:\Program Files\Anaconda3\envs\yolov5-v4.0\lib\site-packages\torch\nn\modules\loss.py", line 961, in forward
return F.cross_entropy(input, target, weight=self.weight,
File "C:\Program Files\Anaconda3\envs\yolov5-v4.0\lib\site-packages\torch\nn\functional.py", line 2468, in cross_entropy
return nll_loss(log_softmax(input, 1), target, weight, None, ignore_index, None, reduction)
File "C:\Program Files\Anaconda3\envs\yolov5-v4.0\lib\site-packages\torch\nn\functional.py", line 2264, in nll_loss
ret = torch._C._nn.nll_loss(input, target.long(), weight, _Reduction.get_enum(reduction), ignore_index) # target
RuntimeError: multi-target not supported at C:/w/b/windows/pytorch/aten/src\THCUNN/generic/ClassNLLCriterion.cu:15
i fixed "loss = loss_fn(output, target)" to "loss = loss_fn(output, target.squeeze())", this error still appears
what can i do?

from volo.

Do you use token label?
If not, maybe you should modify "return_dense=True, mix_token=True" to "return_dense=False, mix_token=False" in

from volo.