search

Project-MONAI / model-zoo

MONAI Model Zoo that hosts models in the MONAI Bundle format.
Apache License 2.0
204 stars 69 forks source link

In the project "lung_nodule_ct_detection", how to add the configuration for resuming training and loading a pre-trained model.pt? #712

Closed dqy818 closed 1 week ago

dqy818 commented 2 weeks ago

In the project "lung_nodule_ct_detection", how to add the configuration for resuming training and loading a pre-trained model.pt?

yiheng-wang-nv commented 2 weeks ago

Hi @dqy818 , I think you can refer to vista3d: https://github.com/Project-MONAI/model-zoo/blob/dev/models/vista3d/configs/train.json#L221 add a checkpoint loader

dqy818 commented 1 week ago

thank you