search

HealthML / self-supervised-3d-tasks

Apache License 2.0
187 stars 40 forks source link

Fine-tuning model: where is the model saved at #20

Closed lokeycookie closed 2 years ago

lokeycookie commented 2 years ago

Hi, I was reading the research paper and saw that 3D-Unet is eventually trained and evaluated on different sizes of labelled data. After running the pre-training file and fine-tuning file in this repo, where will the model (eg 3D-unet) be saved at? Do we need to specify the path to where this fine-tuned model is saved?

aihamtaleb commented 2 years ago

Hi, the checkpoint files are saved in the working_dir: https://github.com/HealthML/self-supervised-3d-tasks/blob/9f21099867f3ac10fe232bd60899483079fd7c91/self_supervised_3d_tasks/train.py#L51

this is an argument you can set easily for the train function

aihamtaleb commented 2 years ago

Please notify us if you have further questions. Closing this for now.