Good day,
I am trying to reproduce some results of this model and the memory consumption is crazy (40gb for 256*256, batch size =3), and the computation time is also very slow, although the number of parameters is very low (~300k). My suggestion that maybe on newer version some changes were done so the code from the older version doesn’t work properly on a new pytorch.
Unfortunately I can’t check the code on an older version because I have CUDA 11 and as far as I understand only pytorch 1,7 is compatible with it.
I am wondering is it possible that on a new version the code runs much slower and consumes a lot of memory? If so, what is the possible reason for that and how it can be fixed?
I also do not exclude the fact the this model has complex connections or wasn’t implemented correctly and simply doesn’t work.