How to continue training the interrupted model

YanzuoLu / CFLD

[CVPR 2024 Highlight] Coarse-to-Fine Latent Diffusion for Pose-Guided Person Image Synthesis

MIT License

183 stars 12 forks source link

Closed Xuzhenhao186 closed 5 months ago

Xuzhenhao186 commented 6 months ago

Excuse me, If there is an interruption during training, how do I continue training

YanzuoLu commented 5 months ago

Hi, @Xuzhenhao186 You should try accelerator.load_state api to load the ckpt save from accelerator.save_state api.