search

yxgeee / FD-GAN

[NeurIPS-2018] FD-GAN: Pose-guided Feature Distilling GAN for Robust Person Re-identification.
https://yxgeee.github.io/projects/fdgan.html
281 stars 80 forks source link

int() argument must be a string, a bytes-like object or a number, not 'tuple' #13

Closed LuoPeng-CV closed 5 years ago

LuoPeng-CV commented 5 years ago

Hello, when I try to pretrain the baseline in stage1,I got a problem.Here is the error imformation, Traceback (most recent call last): File "train.py", line 121, in <module> main() File "train.py", line 75, in main for i, data in enumerate(train_loader): File "/raid/home/kml/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 451, in __iter__ return _DataLoaderIter(self) File "/raid/home/kml/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 247, in __init__ self._put_indices() File "/raid/home/kml/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 295, in _put_indices indices = next(self.sample_iter, None) File "/raid/home/kml/anaconda3/lib/python3.6/site-packages/torch/utils/data/sampler.py", line 139, in __iter__ batch.append(int(idx)) TypeError: int() argument must be a string, a bytes-like object or a number, not 'tuple'

And I download the pretrained models,but when I try to test them,I got the same problem.

I will deeply appreciate it if you could give some advices. Thank you!

yang502 commented 5 years ago

@Rocccc Excuse me. How do you solve this problem?