Some questions about the pytorch version

[ewrfcas]

Results from pytorch1.9 have more disorderly lines compared with ones from pytorch1.3.1. Are there any bugs for it? Lines from pytorch1.9:

Lines from pytorch1.3.1:

My HAWP model was trained in pytorch1.3.1

[cherubicXN]

Yes, it is. It is due to that PyTorch changed the behavior of the division function. I remembered that I fixed the bug and you can update your code. For the old version, the performance will be dramatically dropped from 66.5 to 40+ for the official model.

[ewrfcas]

Thanks for your reply. I have tried to change the code. From y = (index / width).float() + torch.gather(joff[1], 0, index) + 0.5 to y = (index // width).float() + torch.gather(joff[1], 0, index) + 0.5. But this question is not solved...

Any other codes should be changed?

[cherubicXN]

Have you trained the model again after fixing the bug?

[ewrfcas]

I have retrained the model before fixing it.

[cherubicXN]

I have retrained the model before fixing it.

How about the evaluation results?

[ewrfcas]

I added some extra data augmentation, and the results are slightly better than the paper reported in torch1.3.1. But I have not tested them for quantitative metrics in torch1.9.

[cherubicXN]

I added some extra data augmentation, and the results are slightly better than the paper reported in torch1.3.1. But I have not tested them for quantitative metrics in torch1.9.

Maybe some other changes in PyTorch lead to your results. In recent days, I have extensively trained many models with PyTorch-1.10. I did not have observed any incorrect results on my side after fixing the abovementioned bug as you did.

[ewrfcas]

Thanks. I think that the model needs to be retrained in a new PyTorch version.

[cherubicXN]

Yes, it should be retrained.