search

shuaichaochao / HybridHash

HybridHash: Hybrid Convolutional and Self-Attention Deep Hashing for Image Retrieval (ICMR 2024)
8 stars 1 forks source link

Training details #4

Open lwd01 opened 1 month ago

lwd01 commented 1 month ago

Hello author, when I reproduced the CIFAR dataset, I found that the loss kept hovering at 0.54, and then the test found that the map value was about 0.45, so I would like to ask the details of the training here, and is the pre-training weight from my own training?

shuaichaochao commented 1 week ago

The training parameters are the same as in the train.py file, and the download link for the pre-training weights is on the fourteenth line in the HybridHash.py file, please make sure to load it.