search

guoqincode / Open-AnimateAnyone

Unofficial Implementation of Animate Anyone
2.9k stars 233 forks source link

model saving #8

Closed Carlyx closed 9 months ago

Carlyx commented 9 months ago

Hi, it seems that you train the 2D unet, referencenet, and poseguider during the first stage, but you don't save parameters of 2D unet.

guoqincode commented 9 months ago

I didn't train 2d unet, I set unet.requiresgrad(False)

Carlyx commented 9 months ago

You set unet.requiresgrad(False) in (https://github.com/guoqincode/AnimateAnyone-unofficial/blob/45ecbe2d8495c9663e213e0d7dd2df06ca5bcd33/train.py#L224C1-L224C1), but the grad is set to True in the next few lines (https://github.com/guoqincode/AnimateAnyone-unofficial/blob/45ecbe2d8495c9663e213e0d7dd2df06ca5bcd33/train.py#L228). I print the grad and it shows True.

guoqincode commented 9 months ago

Sorry, I forgot to upload the latest config and train.py on github. . .