monailabel.tasks.train.bundle module

class monailabel.tasks.train.bundle.BundleConstants[source]

Bases: object

configs()[source]
Return type

Sequence[str]

key_bundle_root()[source]
Return type

str

key_device()[source]
Return type

str

key_displayable_configs()[source]
Return type

Sequence[str]

key_experiment_name()[source]
Return type

str

key_network()[source]
Return type

str

key_network_def()[source]
Return type

str

key_run_name()[source]
Return type

str

key_tracking()[source]
Return type

str

key_tracking_uri()[source]
Return type

str

key_train_dataset_data()[source]
Return type

str

key_train_handlers()[source]
Return type

str

key_train_trainer_max_epochs()[source]
Return type

str

key_validate_dataset_data()[source]
Return type

str

metadata_json()[source]
Return type

str

model_pytorch()[source]
Return type

str

multi_gpu_configs()[source]
Return type

Sequence[str]

class monailabel.tasks.train.bundle.BundleTrainTask(path, conf, const=None, enable_tracking=False, model_dict_key='model', load_strict=False)[source]

Bases: monailabel.interfaces.tasks.train.TrainTask

config()[source]
info()[source]
is_valid()[source]
run_multi_gpu(request, cmd, env)[source]
run_single_gpu(request, overrides)[source]