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

The last model cannot be generated #23

Open Haoru opened 5 years ago

Haoru commented 5 years ago

I encountered the following problem when I trained the III step. It has been trained, but the model can't be saved. Could you give me some help?

Traceback (most recent call last): File "train.py", line 117, in main() File "train.py", line 96, in main mAP = evaluator.evaluate(val_loader, dataset.val, dataset.val, top1=False) NameError: name 'val_loader' is not defined

liulitianji commented 5 years ago

Hello, I encounter the same problem. Have you solved it?

yxgeee commented 5 years ago

It is a mistake that I missed the val_loader when re-arranging the code. You could follow https://github.com/yxgeee/FD-GAN/blob/master/baseline.py#L59 to create the val loader. @liulitianji @Haoru

Haoru commented 5 years ago

The problem has been solved. I missed the val_loader in my training. Thank you very much~