JointEmbedding icon indicating copy to clipboard operation
JointEmbedding copied to clipboard

Can not reproduce the results in the paper when training from scratch

Open florianlanger opened this issue 1 year ago • 1 comments

Hi Manuel, thanks for your work. I am having trouble to reproduce the retrieval accuracies (or ranking accuracies) quoted in the paper and in issue #4. I am training the network from scratch with the provided code (without any modifications) on the provided preprocessed data. The results are so far off that it seems like there must be a bug somewhere.

I get the following total retrieval accuracies after the respective number of train steps. Number steps: 2k , 25 k, 50 k , 75 k , 100 k Orig settings repo Retrieval accuracy: 0.03, 0.09, 0.10, 0.08, 0.03 Mod settings Retrieval accuracy: 0.01 , 0.03, 0.03, 0.05 , 0.05

For the mod settings in line with the paper I set the triplet margin to 0.2 (as opposed to 0.01) and change the learning rate scheduler to update every 20 k steps as opposed to 40 k steps. As far as I can tell all other settings in the repo match the information provided in the paper.

The result quoted in the paper is 0.43 and in issue #4 it is 0.42. I know this is fairly old code but would be amazing if either you or @TimFelixBeyer could look into this. Thanks a lot! Florian

florianlanger avatar Oct 23 '23 12:10 florianlanger

See email. I'll leave it open until we have resolved it.

xheon avatar Oct 24 '23 22:10 xheon