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

NameError:name 'val_loader' is not defined #38

Closed haozhihui110 closed 4 years ago

haozhihui110 commented 4 years ago

Traceback(most recent call last): File "train.py" , line in 119, in main() File "train.py" , line in 97, in main mAP = evaluator.evaluate(val_loader, dataset.val, dataset.val, top1=False)

NameError:name 'val_loader' is not defined

yxgeee commented 4 years ago

You could simply change “val_loader” to “test_loader” for fixing this error, since I forget to define the val_loader in this version of code.

yxgeee commented 4 years ago

Or you can define the val_loader by adding 

    val_loader = DataLoader(
        Preprocessor(dataset.val,
                     root=dataset.images_dir, transform=test_transformer),
        batch_size=batch_size, num_workers=workers,
        shuffle=False, pin_memory=False)
haozhihui110 commented 4 years ago

Or you can define the val_loader by adding

    val_loader = DataLoader(
        Preprocessor(dataset.val,
                     root=dataset.images_dir, transform=test_transformer),
        batch_size=batch_size, num_workers=workers,
        shuffle=False, pin_memory=False)

alright, thank you very much