Cysu / open-reid

Open source person re-identification library in python
https://cysu.github.io/open-reid/
MIT License
1.33k stars 349 forks source link

RuntimeError: zero-dimensional tensor (at position 0) cannot be concatenated #93

Closed WANGYYYYYYYYY closed 4 years ago

WANGYYYYYYYYY commented 4 years ago

python3 triplet_loss.py -d market1501 -a resnet50 --combine-trainval --logs-dir logs/triplet-loss/market1501-resnet50

when I use this command to run open-reid, I met the problem:

Traceback (most recent call last): File "triplet_loss.py", line 220, in main(parser.parse_args()) File "triplet_loss.py", line 144, in main trainer.train(epoch, train_loader, optimizer) File "/opt/data/private/open-reid/reid/trainers.py", line 31, in train loss, prec1 = self._forward(inputs, targets) File "/opt/data/private/open-reid/reid/trainers.py", line 80, in _forward loss, prec = self.criterion(outputs, targets) File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 547, in call result = self.forward(*input, **kwargs) File "/opt/data/private/open-reid/reid/loss/triplet.py", line 27, in forward dist_ap = torch.cat(dist_ap) RuntimeError: zero-dimensional tensor (at position 0) cannot be concatenated

I need the help, thank you very much.!