Compatibility with PyTorch 1.5 and later

henzler / neuraltexture

Learning a Neural 3D Texture Space from 2D Exemplars [CVPR 2020]

https://geometry.cs.ucl.ac.uk/projects/2020/neuraltexture/

MIT License

108 stars 20 forks source link

Compatibility with PyTorch 1.5 and later #7

Open madhawav opened 3 years ago

madhawav commented 3 years ago

Hi @henzler,

Have you tested training new models on PyTorch versions newer than 1.4?

I am doing a follow-up work based on this project and I have noticed that the model training performance degrades noticeably when I train in PyTorch 1.5 or later. Inference on models trained using PyTorch 1.4, via recent versions of PyTorch works just fine. I am guessing this issue is related to the noise operator implementation I obtained from this codebase.

henzler commented 3 years ago

Unfortunately, I have not tested if it works with a version newer than 1.4.