mridc.core package
Subpackages
- mridc.core.classes package
- Submodules
- mridc.core.classes.common module
- mridc.core.classes.dataset module
- mridc.core.classes.export module
ExportFormat
Exportable
Exportable.disabled_deployment_input_names
Exportable.disabled_deployment_output_names
Exportable.export()
Exportable.get_export_subnet()
Exportable.input_module
Exportable.input_names
Exportable.list_export_subnets()
Exportable.output_module
Exportable.output_names
Exportable.supported_export_formats
- mridc.core.classes.loss module
- mridc.core.classes.modelPT module
ModelPT
ModelPT.__init_subclass__()
ModelPT.cfg
ModelPT.configure_optimizers()
ModelPT.extract_state_dict_from()
ModelPT.get_test_dataloader_prefix()
ModelPT.get_validation_dataloader_prefix()
ModelPT.load_from_checkpoint()
ModelPT.load_part_of_state_dict()
ModelPT.maybe_init_from_pretrained_checkpoint()
ModelPT.multi_test_epoch_end()
ModelPT.multi_validation_epoch_end()
ModelPT.num_weights
ModelPT.prepare_test()
ModelPT.register_artifact()
ModelPT.restore_from()
ModelPT.save_to()
ModelPT.set_trainer()
ModelPT.set_world_size()
ModelPT.setup_multiple_test_data()
ModelPT.setup_multiple_validation_data()
ModelPT.setup_optimization()
ModelPT.setup_optimizer_param_groups()
ModelPT.setup_test_data()
ModelPT.setup_training_data()
ModelPT.setup_validation_data()
ModelPT.summarize()
ModelPT.teardown()
ModelPT.test_dataloader()
ModelPT.test_epoch_end()
ModelPT.train_dataloader()
ModelPT.training
ModelPT.update_save_restore_connector()
ModelPT.val_dataloader()
ModelPT.validation_epoch_end()
- mridc.core.classes.module module
- Module contents
- mridc.core.conf package
- Submodules
- mridc.core.conf.base_config module
- mridc.core.conf.dataloader module
DataLoaderConfig
DataLoaderConfig.batch_sampler
DataLoaderConfig.batch_size
DataLoaderConfig.collate_fn
DataLoaderConfig.drop_last
DataLoaderConfig.multiprocessing_context
DataLoaderConfig.num_workers
DataLoaderConfig.pin_memory
DataLoaderConfig.sampler
DataLoaderConfig.shuffle
DataLoaderConfig.timeout
DataLoaderConfig.worker_init_fn
- mridc.core.conf.hydra_runner module
- mridc.core.conf.modelPT module
- mridc.core.conf.optimizers module
- mridc.core.conf.schedulers module
CosineAnnealingParams
CyclicLRParams
ExponentialLRParams
InverseSquareRootAnnealingParams
NoamAnnealingParams
NoamHoldAnnealingParams
PolynomialDecayAnnealingParams
PolynomialHoldDecayAnnealingParams
ReduceLROnPlateauParams
SchedulerParams
SquareAnnealingParams
SquareRootAnnealingParams
SquareRootConstantSchedulerParams
StepLRParams
WarmupAnnealingHoldSchedulerParams
WarmupAnnealingParams
WarmupHoldSchedulerParams
WarmupSchedulerParams
get_scheduler_config()
register_scheduler_params()
- mridc.core.conf.trainer module
TrainerConfig
TrainerConfig.accelerator
TrainerConfig.accumulate_grad_batches
TrainerConfig.amp_backend
TrainerConfig.amp_level
TrainerConfig.auto_lr_find
TrainerConfig.auto_scale_batch_size
TrainerConfig.auto_select_gpus
TrainerConfig.benchmark
TrainerConfig.callbacks
TrainerConfig.check_val_every_n_epoch
TrainerConfig.default_root_dir
TrainerConfig.detect_anomaly
TrainerConfig.deterministic
TrainerConfig.devices
TrainerConfig.enable_checkpointing
TrainerConfig.enable_model_summary
TrainerConfig.enable_progress_bar
TrainerConfig.fast_dev_run
TrainerConfig.gpus
TrainerConfig.gradient_clip_algorithm
TrainerConfig.gradient_clip_val
TrainerConfig.ipus
TrainerConfig.limit_predict_batches
TrainerConfig.limit_test_batches
TrainerConfig.limit_train_batches
TrainerConfig.limit_val_batches
TrainerConfig.log_every_n_steps
TrainerConfig.logger
TrainerConfig.max_epochs
TrainerConfig.max_steps
TrainerConfig.max_time
TrainerConfig.min_epochs
TrainerConfig.min_steps
TrainerConfig.move_metrics_to_cpu
TrainerConfig.multiple_trainloader_mode
TrainerConfig.num_nodes
TrainerConfig.num_sanity_val_steps
TrainerConfig.overfit_batches
TrainerConfig.plugins
TrainerConfig.precision
TrainerConfig.profiler
TrainerConfig.reload_dataloaders_every_n_epochs
TrainerConfig.replace_sampler_ddp
TrainerConfig.resume_from_checkpoint
TrainerConfig.strategy
TrainerConfig.sync_batchnorm
TrainerConfig.tpu_cores
TrainerConfig.track_grad_norm
TrainerConfig.val_check_interval
TrainerConfig.weights_save_path
- Module contents
- mridc.core.connectors package
- Submodules
- mridc.core.connectors.save_restore_connector module
SaveRestoreConnector
SaveRestoreConnector.extract_state_dict_from()
SaveRestoreConnector.load_config_and_state_dict()
SaveRestoreConnector.load_instance_with_state_dict()
SaveRestoreConnector.model_config_yaml
SaveRestoreConnector.model_extracted_dir
SaveRestoreConnector.model_weights_ckpt
SaveRestoreConnector.register_artifact()
SaveRestoreConnector.restore_from()
SaveRestoreConnector.save_to()
- Module contents
- mridc.core.neural_types package
- Submodules
- mridc.core.neural_types.axes module
- mridc.core.neural_types.comparison module
NeuralTypeComparisonResult
NeuralTypeComparisonResult.CONTAINER_SIZE_MISMATCH
NeuralTypeComparisonResult.DIM_INCOMPATIBLE
NeuralTypeComparisonResult.GREATER
NeuralTypeComparisonResult.INCOMPATIBLE
NeuralTypeComparisonResult.LESS
NeuralTypeComparisonResult.SAME
NeuralTypeComparisonResult.SAME_TYPE_INCOMPATIBLE_PARAMS
NeuralTypeComparisonResult.TRANSPOSE_SAME
NeuralTypeComparisonResult.UNCHECKED
- mridc.core.neural_types.elements module
CategoricalValuesType
ChannelType
ElementType
FloatType
ImageFeatureValue
ImageValue
Index
IntType
LabelsType
Length
LengthsType
LogDeterminantType
LogprobsType
LossType
MRISignal
MaskType
NormalDistributionLogVarianceType
NormalDistributionMeanType
NormalDistributionSamplesType
NormalizedImageValue
PredictionsType
ProbsType
ReconstructionTarget
RecurrentsType
RegressionValuesType
SequenceToSequenceAlignmentType
StringLabel
StringType
Target
VoidType
- mridc.core.neural_types.neural_type module
- Module contents
- mridc.core.optim package
- Submodules
- mridc.core.optim.adafactor module
- mridc.core.optim.lr_scheduler module
CosineAnnealing
InverseSquareRootAnnealing
NoamAnnealing
NoamHoldAnnealing
PolynomialDecayAnnealing
PolynomialHoldDecayAnnealing
SquareAnnealing
SquareRootAnnealing
SquareRootConstantPolicy
T5InverseSquareRootAnnealing
WarmupAnnealHoldPolicy
WarmupAnnealing
WarmupHoldPolicy
WarmupPolicy
compute_max_steps()
get_scheduler()
prepare_lr_scheduler()
register_scheduler()
- mridc.core.optim.novograd module
- mridc.core.optim.optimizer_with_master_params module
GradBucket
MainParamsOptimizerWrapper
MainParamsOptimizerWrapper.allreduce_main_grads()
MainParamsOptimizerWrapper.async_master_grads_allreduce
MainParamsOptimizerWrapper.copy_model_grads_to_main_grads()
MainParamsOptimizerWrapper.defaults
MainParamsOptimizerWrapper.fp32_grad_accumulation
MainParamsOptimizerWrapper.get_parameters()
MainParamsOptimizerWrapper.load_state_dict()
MainParamsOptimizerWrapper.no_sync()
MainParamsOptimizerWrapper.param_groups
MainParamsOptimizerWrapper.reload_model_params()
MainParamsOptimizerWrapper.state
MainParamsOptimizerWrapper.state_dict()
MainParamsOptimizerWrapper.step()
MainParamsOptimizerWrapper.zero_grad()
- mridc.core.optim.optimizers module
- Module contents
- mridc.core.utils package