MONet_convert_ckpt_to_ts script

Convert nnUNet checkpoint to TorchScript

usage: MONet_convert_ckpt_to_ts [-h] --bundle_root BUNDLE_ROOT
                                [--checkpoint_name CHECKPOINT_NAME]
                                [--nnunet_trainer_name NNUNET_TRAINER_NAME]
                                [--fold FOLD]
                                [--dataset_name_or_id DATASET_NAME_OR_ID]

Named Arguments

--bundle_root

Path to the nnUNet bundle root

--checkpoint_name

Checkpoint name

Default: “model.pt”

--nnunet_trainer_name

Trainer name

Default: “nnUNetTrainer”

--fold

Fold number

Default: 0

--dataset_name_or_id

Dataset name or ID to convert

Default: “00”