search

tobias-kirschstein / nersemble

[Siggraph '23] NeRSemble: Neural Radiance Field Reconstruction of Human Heads
https://tobias-kirschstein.github.io/nersemble/
175 stars 7 forks source link

Resume Training from Checkpoint #10

Closed KaneLindsay closed 3 months ago

KaneLindsay commented 3 months ago

Is there functionality to load a previous model checkpoint to continue training?

tobias-kirschstein commented 3 months ago

Hi, thank you for your interest in our work. I added a --resume_run flag to the train script. If provided, it will try to load the latest checkpoint from the specified run. For example --resume_run NERS-9018 would continue training the pretrained model for participant ID 18. If you wish to resume a different checkpoint than the latest, you can use the --resume_checkpoint flag. Hope this helps, and tell me if you encounter any issues. I haven't tested this extensively.

KaneLindsay commented 3 months ago

Thank you for your prompt response, I can now successfully load checkpoints for training.