search

MCC-WH / CSA

Official implementation of NeurIPS 2021 paper "Contextual Similarity Aggregation with Self-attention for Visual Re-ranking"
MIT License
25 stars 3 forks source link

training issue: #7

Closed vink1994 closed 2 years ago

vink1994 commented 2 years ago

I am having trouble in training the model; kindly look into it, thanks! Not using distributed mode Traceback (most recent call last): File "D:\Initital_code\CSA-main\CSA-main\train.py", line 174, in main(args=args) File "D:\Initital_code\CSA-main\CSA-main\train.py", line 32, in main dataset_names = list(eval(args.dataset_names)) File "", line 1, in NameError: name 'rSfM120k' is not defined

MCC-WH commented 2 years ago

I have updated the code a few days ago, you can re-download and try.

vink1994 commented 2 years ago

I tried with updated code; getting same error

MCC-WH commented 2 years ago

The current code works fine on my machine, you need to check it out yourself.