HealthML / active-segmentation

ActiveSegmentation: A Simulation Framework for Benchmarking Active Learning Strategies for 3D Medical Image Segmentation
GNU Affero General Public License v3.0
16 stars 1 forks source link

Add logging models as wandb artifacts #61

Closed Jasperhino closed 2 years ago

Jasperhino commented 2 years ago

closes #61

This will create an artefact for each checkpoint during training, named model-run_id:version. E.g. "model-15xnat09:v2" We could then download and use these models with wandb.run.use_artifact("model-15xnat09:v2")