报错内容:
Traceback (most recent call last):
File "main.py", line 48, in <module>
training_model.train()
File "/home/usr/longtail/run_networks.py", line 205, in train
self.batch_loss(labels)
File "/home/usr/longtail/run_networks.py", line 148, in batch_loss
self.loss_perf = self.criterions['PerformanceLoss'](self.logits, labels) \
File "/home/usr/.pyenv/versions/py36torch/lib/python3.6/site-packages/torch/nn/modules/module.py", line 722, in _call_impl
result = self.forward(*input, **kwargs)
File "/home/usr/.pyenv/versions/py36torch/lib/python3.6/site-packages/torch/nn/modules/loss.py", line 948, in forward
ignore_index=self.ignore_index, reduction=self.reduction)
File "/home/usr/.pyenv/versions/py36torch/lib/python3.6/site-packages/torch/nn/functional.py", line 2422, in cross_entropy
return nll_loss(log_softmax(input, 1), target, weight, None, ignore_i