nashory / DeLF-pytorch

PyTorch Implementation of "Large-Scale Image Retrieval with Attentive Deep Local Features"
MIT License
343 stars 63 forks source link

train PCA #17

Closed ghost closed 4 years ago

ghost commented 5 years ago

error

fatemeakbari commented 4 years ago

Hello, I have the same problem, do you find a solution?

fatemeakbari commented 4 years ago

Thanks a lot

fatemeakbari commented 4 years ago

Hello, I didn't continue, but one idea that may help you, use of "iter" over your dataloader instead loop over dataloader in extractor.py file.

fatemeakbari commented 4 years ago

Hello, I train on a custom dataset