How to load a checkpoint to resume train

yuhaoliu7456 / Diff-Plugin

[CVPR 2024] Official code release of our paper "Diff-Plugin: Revitalizing Details for Diffusion-based Low-level tasks"

130 stars 4 forks source link

How to load a checkpoint to resume train #18

Open Zerycii opened 1 week ago

Zerycii commented 1 week ago

Thanks for your eminent work but i met a problem. How to load the checkpoint for training? my coda is : --resume_from_checkpoint="/home/lmh/lzq/Diff-Plugin/pre-trained/$task" \ This task is demoire.Thanks for your generous help.