mridc.core package
Subpackages
- mridc.core.classes package
- Submodules
- mridc.core.classes.common module
- mridc.core.classes.dataset module
- mridc.core.classes.export module
ExportFormatExportableExportable.disabled_deployment_input_namesExportable.disabled_deployment_output_namesExportable.export()Exportable.get_export_subnet()Exportable.input_moduleExportable.input_namesExportable.list_export_subnets()Exportable.output_moduleExportable.output_namesExportable.supported_export_formats
- mridc.core.classes.loss module
- mridc.core.classes.modelPT module
ModelPTModelPT.__init_subclass__()ModelPT.cfgModelPT.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_weightsModelPT.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.trainingModelPT.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
DataLoaderConfigDataLoaderConfig.batch_samplerDataLoaderConfig.batch_sizeDataLoaderConfig.collate_fnDataLoaderConfig.drop_lastDataLoaderConfig.multiprocessing_contextDataLoaderConfig.num_workersDataLoaderConfig.pin_memoryDataLoaderConfig.samplerDataLoaderConfig.shuffleDataLoaderConfig.timeoutDataLoaderConfig.worker_init_fn
- mridc.core.conf.hydra_runner module
- mridc.core.conf.modelPT module
- mridc.core.conf.optimizers module
- mridc.core.conf.schedulers module
CosineAnnealingParamsCyclicLRParamsExponentialLRParamsInverseSquareRootAnnealingParamsNoamAnnealingParamsNoamHoldAnnealingParamsPolynomialDecayAnnealingParamsPolynomialHoldDecayAnnealingParamsReduceLROnPlateauParamsSchedulerParamsSquareAnnealingParamsSquareRootAnnealingParamsSquareRootConstantSchedulerParamsStepLRParamsWarmupAnnealingHoldSchedulerParamsWarmupAnnealingParamsWarmupHoldSchedulerParamsWarmupSchedulerParamsget_scheduler_config()register_scheduler_params()
- mridc.core.conf.trainer module
TrainerConfigTrainerConfig.acceleratorTrainerConfig.accumulate_grad_batchesTrainerConfig.amp_backendTrainerConfig.amp_levelTrainerConfig.auto_lr_findTrainerConfig.auto_scale_batch_sizeTrainerConfig.auto_select_gpusTrainerConfig.benchmarkTrainerConfig.callbacksTrainerConfig.check_val_every_n_epochTrainerConfig.default_root_dirTrainerConfig.detect_anomalyTrainerConfig.deterministicTrainerConfig.devicesTrainerConfig.enable_checkpointingTrainerConfig.enable_model_summaryTrainerConfig.enable_progress_barTrainerConfig.fast_dev_runTrainerConfig.gpusTrainerConfig.gradient_clip_algorithmTrainerConfig.gradient_clip_valTrainerConfig.ipusTrainerConfig.limit_predict_batchesTrainerConfig.limit_test_batchesTrainerConfig.limit_train_batchesTrainerConfig.limit_val_batchesTrainerConfig.log_every_n_stepsTrainerConfig.loggerTrainerConfig.max_epochsTrainerConfig.max_stepsTrainerConfig.max_timeTrainerConfig.min_epochsTrainerConfig.min_stepsTrainerConfig.move_metrics_to_cpuTrainerConfig.multiple_trainloader_modeTrainerConfig.num_nodesTrainerConfig.num_sanity_val_stepsTrainerConfig.overfit_batchesTrainerConfig.pluginsTrainerConfig.precisionTrainerConfig.profilerTrainerConfig.reload_dataloaders_every_n_epochsTrainerConfig.replace_sampler_ddpTrainerConfig.resume_from_checkpointTrainerConfig.strategyTrainerConfig.sync_batchnormTrainerConfig.tpu_coresTrainerConfig.track_grad_normTrainerConfig.val_check_intervalTrainerConfig.weights_save_path
- Module contents
- mridc.core.connectors package
- Submodules
- mridc.core.connectors.save_restore_connector module
SaveRestoreConnectorSaveRestoreConnector.extract_state_dict_from()SaveRestoreConnector.load_config_and_state_dict()SaveRestoreConnector.load_instance_with_state_dict()SaveRestoreConnector.model_config_yamlSaveRestoreConnector.model_extracted_dirSaveRestoreConnector.model_weights_ckptSaveRestoreConnector.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
NeuralTypeComparisonResultNeuralTypeComparisonResult.CONTAINER_SIZE_MISMATCHNeuralTypeComparisonResult.DIM_INCOMPATIBLENeuralTypeComparisonResult.GREATERNeuralTypeComparisonResult.INCOMPATIBLENeuralTypeComparisonResult.LESSNeuralTypeComparisonResult.SAMENeuralTypeComparisonResult.SAME_TYPE_INCOMPATIBLE_PARAMSNeuralTypeComparisonResult.TRANSPOSE_SAMENeuralTypeComparisonResult.UNCHECKED
- mridc.core.neural_types.elements module
CategoricalValuesTypeChannelTypeElementTypeFloatTypeImageFeatureValueImageValueIndexIntTypeLabelsTypeLengthLengthsTypeLogDeterminantTypeLogprobsTypeLossTypeMRISignalMaskTypeNormalDistributionLogVarianceTypeNormalDistributionMeanTypeNormalDistributionSamplesTypeNormalizedImageValuePredictionsTypeProbsTypeReconstructionTargetRecurrentsTypeRegressionValuesTypeSequenceToSequenceAlignmentTypeStringLabelStringTypeTargetVoidType
- 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
CosineAnnealingInverseSquareRootAnnealingNoamAnnealingNoamHoldAnnealingPolynomialDecayAnnealingPolynomialHoldDecayAnnealingSquareAnnealingSquareRootAnnealingSquareRootConstantPolicyT5InverseSquareRootAnnealingWarmupAnnealHoldPolicyWarmupAnnealingWarmupHoldPolicyWarmupPolicycompute_max_steps()get_scheduler()prepare_lr_scheduler()register_scheduler()
- mridc.core.optim.novograd module
- mridc.core.optim.optimizer_with_master_params module
GradBucketMainParamsOptimizerWrapperMainParamsOptimizerWrapper.allreduce_main_grads()MainParamsOptimizerWrapper.async_master_grads_allreduceMainParamsOptimizerWrapper.copy_model_grads_to_main_grads()MainParamsOptimizerWrapper.defaultsMainParamsOptimizerWrapper.fp32_grad_accumulationMainParamsOptimizerWrapper.get_parameters()MainParamsOptimizerWrapper.load_state_dict()MainParamsOptimizerWrapper.no_sync()MainParamsOptimizerWrapper.param_groupsMainParamsOptimizerWrapper.reload_model_params()MainParamsOptimizerWrapper.stateMainParamsOptimizerWrapper.state_dict()MainParamsOptimizerWrapper.step()MainParamsOptimizerWrapper.zero_grad()
- mridc.core.optim.optimizers module
- Module contents
- mridc.core.utils package