search

AliaksandrSiarohin / first-order-model

This repository contains the source code for the paper First Order Motion Model for Image Animation
https://aliaksandrsiarohin.github.io/first-order-model-website/
MIT License
14.55k stars 3.22k forks source link

Higher version pytorch #320

Open Bovey0809 opened 4 years ago

Bovey0809 commented 4 years ago

Thanks for your greate work, I have been playing with it for several days. Now I want to train a cloth dataset where I came across a problem.

I saw the readme.md said the higher version pytorch may behave different. Is it because of the affine function, which will show in the warning "UserWarning: Default upsampling behavior when mode=bilinear is changed to align_corners=False since 0.4.0. Please specify align_corners=True if the old behavior is desired. See the documentation of nn.Upsample for details."?

If I change the settings as the warning suggest, can I use higher version pytorch?

AliaksandrSiarohin commented 4 years ago

No it will not help.

AliaksandrSiarohin commented 4 years ago

Maybe you can try this out, but not sure that it will help. I did not check. https://github.com/AliaksandrSiarohin/first-order-model/issues/146#issue-624354694

jacobwjs commented 3 years ago

Hi Aliaksandr, saw that you closed #146. Has this been tested with training the model on a more recent version of pytorch?

AliaksandrSiarohin commented 3 years ago

I tried reduced training (20 epochs) and it seems to work.

wjtan99 commented 3 years ago

@AliaksandrSiarohin is this fix checked into the current source code? Can we use higher-version Pytorch, like 1.7 now?