Registries

class todd.registries.ClipGradRegistry[source]

Bases: PartialRegistry

data = {'clip_grad_norm_': <function clip_grad_norm_>, 'clip_grad_value_': <function clip_grad_value_>}

class todd.registries.CollateRegistry[source]

Bases: PartialRegistry

data = {}

class todd.registries.ConfigRegistry[source]

Bases: Registry

data = {'py': <class 'todd.configs.py_.PyConfig'>}

class todd.registries.DataLoaderRegistry[source]

Bases: Registry

data = {'DataLoader': <class 'torch.utils.data.dataloader.DataLoader'>, 'PrefetchDataLoader': <class 'todd.patches.torch.utils.PrefetchDataLoader'>}

class todd.registries.DatasetRegistry[source]

Bases: Registry

data = {'COCODataset': <class 'todd.datasets.coco.COCODataset'>, 'ChainDataset': <class 'torch.utils.data.dataset.ChainDataset'>, 'ConcatDataset': <class 'torch.utils.data.dataset.ConcatDataset'>, 'Dataset': <class 'torch.utils.data.dataset.Dataset'>, 'ImageNetDataset': <class 'todd.datasets.imagenet.ImageNetDataset'>, 'IterableDataset': <class 'torch.utils.data.dataset.IterableDataset'>, 'LAIONAestheticsDataset': <class 'todd.datasets.laion_aesthetics.LAIONAestheticsDataset'>, 'LVISDataset': <class 'todd.datasets.lvis.LVISDataset'>, 'Objects365Dataset': <class 'todd.datasets.objects365.Objects365Dataset'>, 'SAMed2DDataset': <class 'todd.datasets.sa_med2d.SAMed2DDataset'>, 'SATINDataset': <class 'todd.datasets.satin.SATINDataset'>, 'StackDataset': <class 'torch.utils.data.dataset.StackDataset'>, 'Subset': <class 'torch.utils.data.dataset.Subset'>, 'TensorDataset': <class 'torch.utils.data.dataset.TensorDataset'>, 'V3DetDataset': <class 'todd.datasets.v3det.V3DetDataset'>}

class todd.registries.InitRegistry[source]

Bases: PartialRegistry

data = {'constant_': <function constant_>, 'dirac_': <function dirac_>, 'eye_': <function eye_>, 'kaiming_normal_': <function kaiming_normal_>, 'kaiming_uniform_': <function kaiming_uniform_>, 'normal_': <function normal_>, 'ones_': <function ones_>, 'orthogonal_': <function orthogonal_>, 'sparse_': <function sparse_>, 'trunc_normal_': <function trunc_normal_>, 'uniform_': <function uniform_>, 'xavier_normal_': <function xavier_normal_>, 'xavier_uniform_': <function xavier_uniform_>, 'zeros_': <function zeros_>}

class todd.registries.InitWeightsMixin[source]

Bases: Module, ABC

abstractmethod init_weights(config)[source]

Parameters:: config (Config)
Return type:: bool

class todd.registries.LRSchedulerRegistry[source]

Bases: Registry

data = {'ChainedScheduler': <class 'torch.optim.lr_scheduler.ChainedScheduler'>, 'ConstantLR': <class 'torch.optim.lr_scheduler.ConstantLR'>, 'CosineAnnealingLR': <class 'torch.optim.lr_scheduler.CosineAnnealingLR'>, 'CosineAnnealingWarmRestarts': <class 'torch.optim.lr_scheduler.CosineAnnealingWarmRestarts'>, 'CyclicLR': <class 'torch.optim.lr_scheduler.CyclicLR'>, 'ExponentialLR': <class 'torch.optim.lr_scheduler.ExponentialLR'>, 'LambdaLR': <class 'torch.optim.lr_scheduler.LambdaLR'>, 'LinearLR': <class 'torch.optim.lr_scheduler.LinearLR'>, 'MultiStepLR': <class 'torch.optim.lr_scheduler.MultiStepLR'>, 'MultiplicativeLR': <class 'torch.optim.lr_scheduler.MultiplicativeLR'>, 'OneCycleLR': <class 'torch.optim.lr_scheduler.OneCycleLR'>, 'PolynomialLR': <class 'torch.optim.lr_scheduler.PolynomialLR'>, 'ReduceLROnPlateau': <class 'torch.optim.lr_scheduler.ReduceLROnPlateau'>, 'SWALR': <class 'torch.optim.swa_utils.SWALR'>, 'SequentialLR': <class 'torch.optim.lr_scheduler.SequentialLR'>, 'StepLR': <class 'torch.optim.lr_scheduler.StepLR'>, '_LRScheduler': <class 'torch.optim.lr_scheduler._LRScheduler'>}

class todd.registries.ModelRegistry[source]

Bases: Registry

data = {'einops_layers_torch_EinMix': <class 'einops.layers.torch.EinMix'>, 'einops_layers_torch_Rearrange': <class 'einops.layers.torch.Rearrange'>, 'einops_layers_torch_Reduce': <class 'einops.layers.torch.Reduce'>, 'todd_patches_torch_nn_ModuleDict': <class 'todd.patches.torch.nn.ModuleDict'>, 'todd_patches_torch_nn_ModuleList': <class 'todd.patches.torch.nn.ModuleList'>, 'todd_patches_torch_nn_Sequential': <class 'todd.patches.torch.nn.Sequential'>, 'todd_registries_model_InitWeightsMixin': <class 'todd.registries.model.InitWeightsMixin'>, 'torch__export_utils__WrappedMethod': <class 'torch._export.utils._WrappedMethod'>, 'torch_ao_nn_intrinsic_modules_fused_BNReLU2d': <class 'torch.ao.nn.intrinsic.modules.fused.BNReLU2d'>, 'torch_ao_nn_intrinsic_modules_fused_BNReLU3d': <class 'torch.ao.nn.intrinsic.modules.fused.BNReLU3d'>, 'torch_ao_nn_intrinsic_modules_fused_ConvAdd2d': <class 'torch.ao.nn.intrinsic.modules.fused.ConvAdd2d'>, 'torch_ao_nn_intrinsic_modules_fused_ConvAddReLU2d': <class 'torch.ao.nn.intrinsic.modules.fused.ConvAddReLU2d'>, 'torch_ao_nn_intrinsic_modules_fused_ConvBn1d': <class 'torch.ao.nn.intrinsic.modules.fused.ConvBn1d'>, 'torch_ao_nn_intrinsic_modules_fused_ConvBn2d': <class 'torch.ao.nn.intrinsic.modules.fused.ConvBn2d'>, 'torch_ao_nn_intrinsic_modules_fused_ConvBn3d': <class 'torch.ao.nn.intrinsic.modules.fused.ConvBn3d'>, 'torch_ao_nn_intrinsic_modules_fused_ConvBnReLU1d': <class 'torch.ao.nn.intrinsic.modules.fused.ConvBnReLU1d'>, 'torch_ao_nn_intrinsic_modules_fused_ConvBnReLU2d': <class 'torch.ao.nn.intrinsic.modules.fused.ConvBnReLU2d'>, 'torch_ao_nn_intrinsic_modules_fused_ConvBnReLU3d': <class 'torch.ao.nn.intrinsic.modules.fused.ConvBnReLU3d'>, 'torch_ao_nn_intrinsic_modules_fused_ConvReLU1d': <class 'torch.ao.nn.intrinsic.modules.fused.ConvReLU1d'>, 'torch_ao_nn_intrinsic_modules_fused_ConvReLU2d': <class 'torch.ao.nn.intrinsic.modules.fused.ConvReLU2d'>, 'torch_ao_nn_intrinsic_modules_fused_ConvReLU3d': <class 'torch.ao.nn.intrinsic.modules.fused.ConvReLU3d'>, 'torch_ao_nn_intrinsic_modules_fused_LinearBn1d': <class 'torch.ao.nn.intrinsic.modules.fused.LinearBn1d'>, 'torch_ao_nn_intrinsic_modules_fused_LinearLeakyReLU': <class 'torch.ao.nn.intrinsic.modules.fused.LinearLeakyReLU'>, 'torch_ao_nn_intrinsic_modules_fused_LinearReLU': <class 'torch.ao.nn.intrinsic.modules.fused.LinearReLU'>, 'torch_ao_nn_intrinsic_modules_fused_LinearTanh': <class 'torch.ao.nn.intrinsic.modules.fused.LinearTanh'>, 'torch_ao_nn_intrinsic_modules_fused__FusedModule': <class 'torch.ao.nn.intrinsic.modules.fused._FusedModule'>, 'torch_ao_nn_intrinsic_qat_modules_conv_fused_ConvBn1d': <class 'torch.ao.nn.intrinsic.qat.modules.conv_fused.ConvBn1d'>, 'torch_ao_nn_intrinsic_qat_modules_conv_fused_ConvBn2d': <class 'torch.ao.nn.intrinsic.qat.modules.conv_fused.ConvBn2d'>, 'torch_ao_nn_intrinsic_qat_modules_conv_fused_ConvBn3d': <class 'torch.ao.nn.intrinsic.qat.modules.conv_fused.ConvBn3d'>, 'torch_ao_nn_intrinsic_qat_modules_conv_fused_ConvBnReLU1d': <class 'torch.ao.nn.intrinsic.qat.modules.conv_fused.ConvBnReLU1d'>, 'torch_ao_nn_intrinsic_qat_modules_conv_fused_ConvBnReLU2d': <class 'torch.ao.nn.intrinsic.qat.modules.conv_fused.ConvBnReLU2d'>, 'torch_ao_nn_intrinsic_qat_modules_conv_fused_ConvBnReLU3d': <class 'torch.ao.nn.intrinsic.qat.modules.conv_fused.ConvBnReLU3d'>, 'torch_ao_nn_intrinsic_qat_modules_conv_fused_ConvReLU1d': <class 'torch.ao.nn.intrinsic.qat.modules.conv_fused.ConvReLU1d'>, 'torch_ao_nn_intrinsic_qat_modules_conv_fused_ConvReLU2d': <class 'torch.ao.nn.intrinsic.qat.modules.conv_fused.ConvReLU2d'>, 'torch_ao_nn_intrinsic_qat_modules_conv_fused_ConvReLU3d': <class 'torch.ao.nn.intrinsic.qat.modules.conv_fused.ConvReLU3d'>, 'torch_ao_nn_intrinsic_qat_modules_conv_fused__ConvBnNd': <class 'torch.ao.nn.intrinsic.qat.modules.conv_fused._ConvBnNd'>, 'torch_ao_nn_intrinsic_qat_modules_linear_fused_LinearBn1d': <class 'torch.ao.nn.intrinsic.qat.modules.linear_fused.LinearBn1d'>, 'torch_ao_nn_intrinsic_qat_modules_linear_relu_LinearReLU': <class 'torch.ao.nn.intrinsic.qat.modules.linear_relu.LinearReLU'>, 'torch_ao_nn_intrinsic_quantized_dynamic_modules_linear_relu_LinearReLU': <class 'torch.ao.nn.intrinsic.quantized.dynamic.modules.linear_relu.LinearReLU'>, 'torch_ao_nn_intrinsic_quantized_modules_bn_relu_BNReLU2d': <class 'torch.ao.nn.intrinsic.quantized.modules.bn_relu.BNReLU2d'>, 'torch_ao_nn_intrinsic_quantized_modules_bn_relu_BNReLU3d': <class 'torch.ao.nn.intrinsic.quantized.modules.bn_relu.BNReLU3d'>, 'torch_ao_nn_intrinsic_quantized_modules_conv_add_ConvAdd2d': <class 'torch.ao.nn.intrinsic.quantized.modules.conv_add.ConvAdd2d'>, 'torch_ao_nn_intrinsic_quantized_modules_conv_add_ConvAddReLU2d': <class 'torch.ao.nn.intrinsic.quantized.modules.conv_add.ConvAddReLU2d'>, 'torch_ao_nn_intrinsic_quantized_modules_conv_relu_ConvReLU1d': <class 'torch.ao.nn.intrinsic.quantized.modules.conv_relu.ConvReLU1d'>, 'torch_ao_nn_intrinsic_quantized_modules_conv_relu_ConvReLU2d': <class 'torch.ao.nn.intrinsic.quantized.modules.conv_relu.ConvReLU2d'>, 'torch_ao_nn_intrinsic_quantized_modules_conv_relu_ConvReLU3d': <class 'torch.ao.nn.intrinsic.quantized.modules.conv_relu.ConvReLU3d'>, 'torch_ao_nn_intrinsic_quantized_modules_linear_relu_LinearLeakyReLU': <class 'torch.ao.nn.intrinsic.quantized.modules.linear_relu.LinearLeakyReLU'>, 'torch_ao_nn_intrinsic_quantized_modules_linear_relu_LinearReLU': <class 'torch.ao.nn.intrinsic.quantized.modules.linear_relu.LinearReLU'>, 'torch_ao_nn_intrinsic_quantized_modules_linear_relu_LinearTanh': <class 'torch.ao.nn.intrinsic.quantized.modules.linear_relu.LinearTanh'>, 'torch_ao_nn_qat_dynamic_modules_linear_Linear': <class 'torch.ao.nn.qat.dynamic.modules.linear.Linear'>, 'torch_ao_nn_qat_modules_conv_Conv1d': <class 'torch.ao.nn.qat.modules.conv.Conv1d'>, 'torch_ao_nn_qat_modules_conv_Conv2d': <class 'torch.ao.nn.qat.modules.conv.Conv2d'>, 'torch_ao_nn_qat_modules_conv_Conv3d': <class 'torch.ao.nn.qat.modules.conv.Conv3d'>, 'torch_ao_nn_qat_modules_conv__ConvNd': <class 'torch.ao.nn.qat.modules.conv._ConvNd'>, 'torch_ao_nn_qat_modules_embedding_ops_Embedding': <class 'torch.ao.nn.qat.modules.embedding_ops.Embedding'>, 'torch_ao_nn_qat_modules_embedding_ops_EmbeddingBag': <class 'torch.ao.nn.qat.modules.embedding_ops.EmbeddingBag'>, 'torch_ao_nn_qat_modules_linear_Linear': <class 'torch.ao.nn.qat.modules.linear.Linear'>, 'torch_ao_nn_quantizable_modules_activation_MultiheadAttention': <class 'torch.ao.nn.quantizable.modules.activation.MultiheadAttention'>, 'torch_ao_nn_quantizable_modules_rnn_LSTM': <class 'torch.ao.nn.quantizable.modules.rnn.LSTM'>, 'torch_ao_nn_quantizable_modules_rnn_LSTMCell': <class 'torch.ao.nn.quantizable.modules.rnn.LSTMCell'>, 'torch_ao_nn_quantizable_modules_rnn__LSTMLayer': <class 'torch.ao.nn.quantizable.modules.rnn._LSTMLayer'>, 'torch_ao_nn_quantizable_modules_rnn__LSTMSingleLayer': <class 'torch.ao.nn.quantizable.modules.rnn._LSTMSingleLayer'>, 'torch_ao_nn_quantized_dynamic_modules_conv_Conv1d': <class 'torch.ao.nn.quantized.dynamic.modules.conv.Conv1d'>, 'torch_ao_nn_quantized_dynamic_modules_conv_Conv2d': <class 'torch.ao.nn.quantized.dynamic.modules.conv.Conv2d'>, 'torch_ao_nn_quantized_dynamic_modules_conv_Conv3d': <class 'torch.ao.nn.quantized.dynamic.modules.conv.Conv3d'>, 'torch_ao_nn_quantized_dynamic_modules_conv_ConvTranspose1d': <class 'torch.ao.nn.quantized.dynamic.modules.conv.ConvTranspose1d'>, 'torch_ao_nn_quantized_dynamic_modules_conv_ConvTranspose2d': <class 'torch.ao.nn.quantized.dynamic.modules.conv.ConvTranspose2d'>, 'torch_ao_nn_quantized_dynamic_modules_conv_ConvTranspose3d': <class 'torch.ao.nn.quantized.dynamic.modules.conv.ConvTranspose3d'>, 'torch_ao_nn_quantized_dynamic_modules_linear_Linear': <class 'torch.ao.nn.quantized.dynamic.modules.linear.Linear'>, 'torch_ao_nn_quantized_dynamic_modules_rnn_GRU': <class 'torch.ao.nn.quantized.dynamic.modules.rnn.GRU'>, 'torch_ao_nn_quantized_dynamic_modules_rnn_GRUCell': <class 'torch.ao.nn.quantized.dynamic.modules.rnn.GRUCell'>, 'torch_ao_nn_quantized_dynamic_modules_rnn_LSTM': <class 'torch.ao.nn.quantized.dynamic.modules.rnn.LSTM'>, 'torch_ao_nn_quantized_dynamic_modules_rnn_LSTMCell': <class 'torch.ao.nn.quantized.dynamic.modules.rnn.LSTMCell'>, 'torch_ao_nn_quantized_dynamic_modules_rnn_PackedParameter': <class 'torch.ao.nn.quantized.dynamic.modules.rnn.PackedParameter'>, 'torch_ao_nn_quantized_dynamic_modules_rnn_RNNBase': <class 'torch.ao.nn.quantized.dynamic.modules.rnn.RNNBase'>, 'torch_ao_nn_quantized_dynamic_modules_rnn_RNNCell': <class 'torch.ao.nn.quantized.dynamic.modules.rnn.RNNCell'>, 'torch_ao_nn_quantized_dynamic_modules_rnn_RNNCellBase': <class 'torch.ao.nn.quantized.dynamic.modules.rnn.RNNCellBase'>, 'torch_ao_nn_quantized_modules_DeQuantize': <class 'torch.ao.nn.quantized.modules.DeQuantize'>, 'torch_ao_nn_quantized_modules_Quantize': <class 'torch.ao.nn.quantized.modules.Quantize'>, 'torch_ao_nn_quantized_modules_activation_ELU': <class 'torch.ao.nn.quantized.modules.activation.ELU'>, 'torch_ao_nn_quantized_modules_activation_Hardswish': <class 'torch.ao.nn.quantized.modules.activation.Hardswish'>, 'torch_ao_nn_quantized_modules_activation_LeakyReLU': <class 'torch.ao.nn.quantized.modules.activation.LeakyReLU'>, 'torch_ao_nn_quantized_modules_activation_MultiheadAttention': <class 'torch.ao.nn.quantized.modules.activation.MultiheadAttention'>, 'torch_ao_nn_quantized_modules_activation_PReLU': <class 'torch.ao.nn.quantized.modules.activation.PReLU'>, 'torch_ao_nn_quantized_modules_activation_ReLU6': <class 'torch.ao.nn.quantized.modules.activation.ReLU6'>, 'torch_ao_nn_quantized_modules_activation_Sigmoid': <class 'torch.ao.nn.quantized.modules.activation.Sigmoid'>, 'torch_ao_nn_quantized_modules_activation_Softmax': <class 'torch.ao.nn.quantized.modules.activation.Softmax'>, 'torch_ao_nn_quantized_modules_batchnorm_BatchNorm2d': <class 'torch.ao.nn.quantized.modules.batchnorm.BatchNorm2d'>, 'torch_ao_nn_quantized_modules_batchnorm_BatchNorm3d': <class 'torch.ao.nn.quantized.modules.batchnorm.BatchNorm3d'>, 'torch_ao_nn_quantized_modules_batchnorm__BatchNorm': <class 'torch.ao.nn.quantized.modules.batchnorm._BatchNorm'>, 'torch_ao_nn_quantized_modules_conv_Conv1d': <class 'torch.ao.nn.quantized.modules.conv.Conv1d'>, 'torch_ao_nn_quantized_modules_conv_Conv2d': <class 'torch.ao.nn.quantized.modules.conv.Conv2d'>, 'torch_ao_nn_quantized_modules_conv_Conv3d': <class 'torch.ao.nn.quantized.modules.conv.Conv3d'>, 'torch_ao_nn_quantized_modules_conv_ConvTranspose1d': <class 'torch.ao.nn.quantized.modules.conv.ConvTranspose1d'>, 'torch_ao_nn_quantized_modules_conv_ConvTranspose2d': <class 'torch.ao.nn.quantized.modules.conv.ConvTranspose2d'>, 'torch_ao_nn_quantized_modules_conv_ConvTranspose3d': <class 'torch.ao.nn.quantized.modules.conv.ConvTranspose3d'>, 'torch_ao_nn_quantized_modules_conv__ConvNd': <class 'torch.ao.nn.quantized.modules.conv._ConvNd'>, 'torch_ao_nn_quantized_modules_conv__ConvTransposeNd': <class 'torch.ao.nn.quantized.modules.conv._ConvTransposeNd'>, 'torch_ao_nn_quantized_modules_dropout_Dropout': <class 'torch.ao.nn.quantized.modules.dropout.Dropout'>, 'torch_ao_nn_quantized_modules_embedding_ops_Embedding': <class 'torch.ao.nn.quantized.modules.embedding_ops.Embedding'>, 'torch_ao_nn_quantized_modules_embedding_ops_EmbeddingBag': <class 'torch.ao.nn.quantized.modules.embedding_ops.EmbeddingBag'>, 'torch_ao_nn_quantized_modules_embedding_ops_EmbeddingPackedParams': <class 'torch.ao.nn.quantized.modules.embedding_ops.EmbeddingPackedParams'>, 'torch_ao_nn_quantized_modules_functional_modules_FXFloatFunctional': <class 'torch.ao.nn.quantized.modules.functional_modules.FXFloatFunctional'>, 'torch_ao_nn_quantized_modules_functional_modules_FloatFunctional': <class 'torch.ao.nn.quantized.modules.functional_modules.FloatFunctional'>, 'torch_ao_nn_quantized_modules_functional_modules_QFunctional': <class 'torch.ao.nn.quantized.modules.functional_modules.QFunctional'>, 'torch_ao_nn_quantized_modules_linear_Linear': <class 'torch.ao.nn.quantized.modules.linear.Linear'>, 'torch_ao_nn_quantized_modules_linear_LinearPackedParams': <class 'torch.ao.nn.quantized.modules.linear.LinearPackedParams'>, 'torch_ao_nn_quantized_modules_normalization_GroupNorm': <class 'torch.ao.nn.quantized.modules.normalization.GroupNorm'>, 'torch_ao_nn_quantized_modules_normalization_InstanceNorm1d': <class 'torch.ao.nn.quantized.modules.normalization.InstanceNorm1d'>, 'torch_ao_nn_quantized_modules_normalization_InstanceNorm2d': <class 'torch.ao.nn.quantized.modules.normalization.InstanceNorm2d'>, 'torch_ao_nn_quantized_modules_normalization_InstanceNorm3d': <class 'torch.ao.nn.quantized.modules.normalization.InstanceNorm3d'>, 'torch_ao_nn_quantized_modules_normalization_LayerNorm': <class 'torch.ao.nn.quantized.modules.normalization.LayerNorm'>, 'torch_ao_nn_quantized_modules_rnn_LSTM': <class 'torch.ao.nn.quantized.modules.rnn.LSTM'>, 'torch_ao_nn_quantized_modules_utils_WeightedQuantizedModule': <class 'torch.ao.nn.quantized.modules.utils.WeightedQuantizedModule'>, 'torch_ao_nn_quantized_reference_modules_conv_Conv1d': <class 'torch.ao.nn.quantized.reference.modules.conv.Conv1d'>, 'torch_ao_nn_quantized_reference_modules_conv_Conv2d': <class 'torch.ao.nn.quantized.reference.modules.conv.Conv2d'>, 'torch_ao_nn_quantized_reference_modules_conv_Conv3d': <class 'torch.ao.nn.quantized.reference.modules.conv.Conv3d'>, 'torch_ao_nn_quantized_reference_modules_conv_ConvTranspose1d': <class 'torch.ao.nn.quantized.reference.modules.conv.ConvTranspose1d'>, 'torch_ao_nn_quantized_reference_modules_conv_ConvTranspose2d': <class 'torch.ao.nn.quantized.reference.modules.conv.ConvTranspose2d'>, 'torch_ao_nn_quantized_reference_modules_conv_ConvTranspose3d': <class 'torch.ao.nn.quantized.reference.modules.conv.ConvTranspose3d'>, 'torch_ao_nn_quantized_reference_modules_conv__ConvNd': <class 'torch.ao.nn.quantized.reference.modules.conv._ConvNd'>, 'torch_ao_nn_quantized_reference_modules_conv__ConvTransposeNd': <class 'torch.ao.nn.quantized.reference.modules.conv._ConvTransposeNd'>, 'torch_ao_nn_quantized_reference_modules_linear_Linear': <class 'torch.ao.nn.quantized.reference.modules.linear.Linear'>, 'torch_ao_nn_quantized_reference_modules_rnn_GRU': <class 'torch.ao.nn.quantized.reference.modules.rnn.GRU'>, 'torch_ao_nn_quantized_reference_modules_rnn_GRUCell': <class 'torch.ao.nn.quantized.reference.modules.rnn.GRUCell'>, 'torch_ao_nn_quantized_reference_modules_rnn_LSTM': <class 'torch.ao.nn.quantized.reference.modules.rnn.LSTM'>, 'torch_ao_nn_quantized_reference_modules_rnn_LSTMCell': <class 'torch.ao.nn.quantized.reference.modules.rnn.LSTMCell'>, 'torch_ao_nn_quantized_reference_modules_rnn_RNNBase': <class 'torch.ao.nn.quantized.reference.modules.rnn.RNNBase'>, 'torch_ao_nn_quantized_reference_modules_rnn_RNNCell': <class 'torch.ao.nn.quantized.reference.modules.rnn.RNNCell'>, 'torch_ao_nn_quantized_reference_modules_rnn_RNNCellBase': <class 'torch.ao.nn.quantized.reference.modules.rnn.RNNCellBase'>, 'torch_ao_nn_quantized_reference_modules_sparse_Embedding': <class 'torch.ao.nn.quantized.reference.modules.sparse.Embedding'>, 'torch_ao_nn_quantized_reference_modules_sparse_EmbeddingBag': <class 'torch.ao.nn.quantized.reference.modules.sparse.EmbeddingBag'>, 'torch_ao_nn_quantized_reference_modules_utils_ReferenceQuantizedModule': <class 'torch.ao.nn.quantized.reference.modules.utils.ReferenceQuantizedModule'>, 'torch_ao_nn_sparse_quantized_dynamic_linear_Linear': <class 'torch.ao.nn.sparse.quantized.dynamic.linear.Linear'>, 'torch_ao_nn_sparse_quantized_linear_Linear': <class 'torch.ao.nn.sparse.quantized.linear.Linear'>, 'torch_ao_nn_sparse_quantized_linear_LinearPackedParams': <class 'torch.ao.nn.sparse.quantized.linear.LinearPackedParams'>, 'torch_ao_quantization__DerivedObserverOrFakeQuantize': <class 'torch.ao.quantization._DerivedObserverOrFakeQuantize'>, 'torch_ao_quantization_fake_quantize_FakeQuantize': <class 'torch.ao.quantization.fake_quantize.FakeQuantize'>, 'torch_ao_quantization_fake_quantize_FakeQuantizeBase': <class 'torch.ao.quantization.fake_quantize.FakeQuantizeBase'>, 'torch_ao_quantization_fake_quantize_FixedQParamsFakeQuantize': <class 'torch.ao.quantization.fake_quantize.FixedQParamsFakeQuantize'>, 'torch_ao_quantization_fake_quantize_FusedMovingAvgObsFakeQuantize': <class 'torch.ao.quantization.fake_quantize.FusedMovingAvgObsFakeQuantize'>, 'torch_ao_quantization_observer_AffineQuantizedObserverBase': <class 'torch.ao.quantization.observer.AffineQuantizedObserverBase'>, 'torch_ao_quantization_observer_FixedQParamsObserver': <class 'torch.ao.quantization.observer.FixedQParamsObserver'>, 'torch_ao_quantization_observer_HistogramObserver': <class 'torch.ao.quantization.observer.HistogramObserver'>, 'torch_ao_quantization_observer_MinMaxObserver': <class 'torch.ao.quantization.observer.MinMaxObserver'>, 'torch_ao_quantization_observer_MovingAverageMinMaxObserver': <class 'torch.ao.quantization.observer.MovingAverageMinMaxObserver'>, 'torch_ao_quantization_observer_MovingAveragePerChannelMinMaxObserver': <class 'torch.ao.quantization.observer.MovingAveragePerChannelMinMaxObserver'>, 'torch_ao_quantization_observer_NoopObserver': <class 'torch.ao.quantization.observer.NoopObserver'>, 'torch_ao_quantization_observer_ObserverBase': <class 'torch.ao.quantization.observer.ObserverBase'>, 'torch_ao_quantization_observer_PerChannelMinMaxObserver': <class 'torch.ao.quantization.observer.PerChannelMinMaxObserver'>, 'torch_ao_quantization_observer_PlaceholderObserver': <class 'torch.ao.quantization.observer.PlaceholderObserver'>, 'torch_ao_quantization_observer_RecordingObserver': <class 'torch.ao.quantization.observer.RecordingObserver'>, 'torch_ao_quantization_observer_ReuseInputObserver': <class 'torch.ao.quantization.observer.ReuseInputObserver'>, 'torch_ao_quantization_observer_UniformQuantizationObserverBase': <class 'torch.ao.quantization.observer.UniformQuantizationObserverBase'>, 'torch_ao_quantization_stubs_DeQuantStub': <class 'torch.ao.quantization.stubs.DeQuantStub'>, 'torch_ao_quantization_stubs_QuantStub': <class 'torch.ao.quantization.stubs.QuantStub'>, 'torch_ao_quantization_stubs_QuantWrapper': <class 'torch.ao.quantization.stubs.QuantWrapper'>, 'torch_export_unflatten_InterpreterModule': <class 'torch.export.unflatten.InterpreterModule'>, 'torch_export_unflatten_InterpreterModuleDispatcher': <class 'torch.export.unflatten.InterpreterModuleDispatcher'>, 'torch_export_unflatten_UnflattenedModule': <class 'torch.export.unflatten.UnflattenedModule'>, 'torch_fx__lazy_graph_module__LazyGraphModule': <class 'torch.fx._lazy_graph_module._LazyGraphModule'>, 'torch_fx_experimental_const_fold_FoldedGraphModule': <class 'torch.fx.experimental.const_fold.FoldedGraphModule'>, 'torch_fx_graph_module_GraphModule': <class 'torch.fx.graph_module.GraphModule'>, 'torch_fx_graph_module__CodeOnlyModule': <class 'torch.fx.graph_module._CodeOnlyModule'>, 'torch_fx_passes_utils_common_HolderModule': <class 'torch.fx.passes.utils.common.HolderModule'>, 'torch_jit__script_RecursiveScriptModule': <class 'torch.jit._script.RecursiveScriptModule'>, 'torch_jit__script_ScriptModule': <class 'torch.jit._script.ScriptModule'>, 'torch_jit__trace_ONNXTracedModule': <class 'torch.jit._trace.ONNXTracedModule'>, 'torch_jit__trace_TopLevelTracedModule': <class 'torch.jit._trace.TopLevelTracedModule'>, 'torch_jit__trace_TracedModule': <class 'torch.jit._trace.TracedModule'>, 'torch_nn_modules_activation_CELU': <class 'torch.nn.modules.activation.CELU'>, 'torch_nn_modules_activation_ELU': <class 'torch.nn.modules.activation.ELU'>, 'torch_nn_modules_activation_GELU': <class 'torch.nn.modules.activation.GELU'>, 'torch_nn_modules_activation_GLU': <class 'torch.nn.modules.activation.GLU'>, 'torch_nn_modules_activation_Hardshrink': <class 'torch.nn.modules.activation.Hardshrink'>, 'torch_nn_modules_activation_Hardsigmoid': <class 'torch.nn.modules.activation.Hardsigmoid'>, 'torch_nn_modules_activation_Hardswish': <class 'torch.nn.modules.activation.Hardswish'>, 'torch_nn_modules_activation_Hardtanh': <class 'torch.nn.modules.activation.Hardtanh'>, 'torch_nn_modules_activation_LeakyReLU': <class 'torch.nn.modules.activation.LeakyReLU'>, 'torch_nn_modules_activation_LogSigmoid': <class 'torch.nn.modules.activation.LogSigmoid'>, 'torch_nn_modules_activation_LogSoftmax': <class 'torch.nn.modules.activation.LogSoftmax'>, 'torch_nn_modules_activation_Mish': <class 'torch.nn.modules.activation.Mish'>, 'torch_nn_modules_activation_MultiheadAttention': <class 'torch.nn.modules.activation.MultiheadAttention'>, 'torch_nn_modules_activation_PReLU': <class 'torch.nn.modules.activation.PReLU'>, 'torch_nn_modules_activation_RReLU': <class 'torch.nn.modules.activation.RReLU'>, 'torch_nn_modules_activation_ReLU': <class 'torch.nn.modules.activation.ReLU'>, 'torch_nn_modules_activation_ReLU6': <class 'torch.nn.modules.activation.ReLU6'>, 'torch_nn_modules_activation_SELU': <class 'torch.nn.modules.activation.SELU'>, 'torch_nn_modules_activation_SiLU': <class 'torch.nn.modules.activation.SiLU'>, 'torch_nn_modules_activation_Sigmoid': <class 'torch.nn.modules.activation.Sigmoid'>, 'torch_nn_modules_activation_Softmax': <class 'torch.nn.modules.activation.Softmax'>, 'torch_nn_modules_activation_Softmax2d': <class 'torch.nn.modules.activation.Softmax2d'>, 'torch_nn_modules_activation_Softmin': <class 'torch.nn.modules.activation.Softmin'>, 'torch_nn_modules_activation_Softplus': <class 'torch.nn.modules.activation.Softplus'>, 'torch_nn_modules_activation_Softshrink': <class 'torch.nn.modules.activation.Softshrink'>, 'torch_nn_modules_activation_Softsign': <class 'torch.nn.modules.activation.Softsign'>, 'torch_nn_modules_activation_Tanh': <class 'torch.nn.modules.activation.Tanh'>, 'torch_nn_modules_activation_Tanhshrink': <class 'torch.nn.modules.activation.Tanhshrink'>, 'torch_nn_modules_activation_Threshold': <class 'torch.nn.modules.activation.Threshold'>, 'torch_nn_modules_adaptive_AdaptiveLogSoftmaxWithLoss': <class 'torch.nn.modules.adaptive.AdaptiveLogSoftmaxWithLoss'>, 'torch_nn_modules_batchnorm_BatchNorm1d': <class 'torch.nn.modules.batchnorm.BatchNorm1d'>, 'torch_nn_modules_batchnorm_BatchNorm2d': <class 'torch.nn.modules.batchnorm.BatchNorm2d'>, 'torch_nn_modules_batchnorm_BatchNorm3d': <class 'torch.nn.modules.batchnorm.BatchNorm3d'>, 'torch_nn_modules_batchnorm_LazyBatchNorm1d': <class 'torch.nn.modules.batchnorm.LazyBatchNorm1d'>, 'torch_nn_modules_batchnorm_LazyBatchNorm2d': <class 'torch.nn.modules.batchnorm.LazyBatchNorm2d'>, 'torch_nn_modules_batchnorm_LazyBatchNorm3d': <class 'torch.nn.modules.batchnorm.LazyBatchNorm3d'>, 'torch_nn_modules_batchnorm_SyncBatchNorm': <class 'torch.nn.modules.batchnorm.SyncBatchNorm'>, 'torch_nn_modules_batchnorm__BatchNorm': <class 'torch.nn.modules.batchnorm._BatchNorm'>, 'torch_nn_modules_batchnorm__LazyNormBase': <class 'torch.nn.modules.batchnorm._LazyNormBase'>, 'torch_nn_modules_batchnorm__NormBase': <class 'torch.nn.modules.batchnorm._NormBase'>, 'torch_nn_modules_channelshuffle_ChannelShuffle': <class 'torch.nn.modules.channelshuffle.ChannelShuffle'>, 'torch_nn_modules_container_Container': <class 'torch.nn.modules.container.Container'>, 'torch_nn_modules_container_ModuleDict': <class 'torch.nn.modules.container.ModuleDict'>, 'torch_nn_modules_container_ModuleList': <class 'torch.nn.modules.container.ModuleList'>, 'torch_nn_modules_container_ParameterDict': <class 'torch.nn.modules.container.ParameterDict'>, 'torch_nn_modules_container_ParameterList': <class 'torch.nn.modules.container.ParameterList'>, 'torch_nn_modules_container_Sequential': <class 'torch.nn.modules.container.Sequential'>, 'torch_nn_modules_conv_Conv1d': <class 'torch.nn.modules.conv.Conv1d'>, 'torch_nn_modules_conv_Conv2d': <class 'torch.nn.modules.conv.Conv2d'>, 'torch_nn_modules_conv_Conv3d': <class 'torch.nn.modules.conv.Conv3d'>, 'torch_nn_modules_conv_ConvTranspose1d': <class 'torch.nn.modules.conv.ConvTranspose1d'>, 'torch_nn_modules_conv_ConvTranspose2d': <class 'torch.nn.modules.conv.ConvTranspose2d'>, 'torch_nn_modules_conv_ConvTranspose3d': <class 'torch.nn.modules.conv.ConvTranspose3d'>, 'torch_nn_modules_conv_LazyConv1d': <class 'torch.nn.modules.conv.LazyConv1d'>, 'torch_nn_modules_conv_LazyConv2d': <class 'torch.nn.modules.conv.LazyConv2d'>, 'torch_nn_modules_conv_LazyConv3d': <class 'torch.nn.modules.conv.LazyConv3d'>, 'torch_nn_modules_conv_LazyConvTranspose1d': <class 'torch.nn.modules.conv.LazyConvTranspose1d'>, 'torch_nn_modules_conv_LazyConvTranspose2d': <class 'torch.nn.modules.conv.LazyConvTranspose2d'>, 'torch_nn_modules_conv_LazyConvTranspose3d': <class 'torch.nn.modules.conv.LazyConvTranspose3d'>, 'torch_nn_modules_conv__ConvNd': <class 'torch.nn.modules.conv._ConvNd'>, 'torch_nn_modules_conv__ConvTransposeMixin': <class 'torch.nn.modules.conv._ConvTransposeMixin'>, 'torch_nn_modules_conv__ConvTransposeNd': <class 'torch.nn.modules.conv._ConvTransposeNd'>, 'torch_nn_modules_distance_CosineSimilarity': <class 'torch.nn.modules.distance.CosineSimilarity'>, 'torch_nn_modules_distance_PairwiseDistance': <class 'torch.nn.modules.distance.PairwiseDistance'>, 'torch_nn_modules_dropout_AlphaDropout': <class 'torch.nn.modules.dropout.AlphaDropout'>, 'torch_nn_modules_dropout_Dropout': <class 'torch.nn.modules.dropout.Dropout'>, 'torch_nn_modules_dropout_Dropout1d': <class 'torch.nn.modules.dropout.Dropout1d'>, 'torch_nn_modules_dropout_Dropout2d': <class 'torch.nn.modules.dropout.Dropout2d'>, 'torch_nn_modules_dropout_Dropout3d': <class 'torch.nn.modules.dropout.Dropout3d'>, 'torch_nn_modules_dropout_FeatureAlphaDropout': <class 'torch.nn.modules.dropout.FeatureAlphaDropout'>, 'torch_nn_modules_dropout__DropoutNd': <class 'torch.nn.modules.dropout._DropoutNd'>, 'torch_nn_modules_flatten_Flatten': <class 'torch.nn.modules.flatten.Flatten'>, 'torch_nn_modules_flatten_Unflatten': <class 'torch.nn.modules.flatten.Unflatten'>, 'torch_nn_modules_fold_Fold': <class 'torch.nn.modules.fold.Fold'>, 'torch_nn_modules_fold_Unfold': <class 'torch.nn.modules.fold.Unfold'>, 'torch_nn_modules_instancenorm_InstanceNorm1d': <class 'torch.nn.modules.instancenorm.InstanceNorm1d'>, 'torch_nn_modules_instancenorm_InstanceNorm2d': <class 'torch.nn.modules.instancenorm.InstanceNorm2d'>, 'torch_nn_modules_instancenorm_InstanceNorm3d': <class 'torch.nn.modules.instancenorm.InstanceNorm3d'>, 'torch_nn_modules_instancenorm_LazyInstanceNorm1d': <class 'torch.nn.modules.instancenorm.LazyInstanceNorm1d'>, 'torch_nn_modules_instancenorm_LazyInstanceNorm2d': <class 'torch.nn.modules.instancenorm.LazyInstanceNorm2d'>, 'torch_nn_modules_instancenorm_LazyInstanceNorm3d': <class 'torch.nn.modules.instancenorm.LazyInstanceNorm3d'>, 'torch_nn_modules_instancenorm__InstanceNorm': <class 'torch.nn.modules.instancenorm._InstanceNorm'>, 'torch_nn_modules_linear_Bilinear': <class 'torch.nn.modules.linear.Bilinear'>, 'torch_nn_modules_linear_Identity': <class 'torch.nn.modules.linear.Identity'>, 'torch_nn_modules_linear_LazyLinear': <class 'torch.nn.modules.linear.LazyLinear'>, 'torch_nn_modules_linear_Linear': <class 'torch.nn.modules.linear.Linear'>, 'torch_nn_modules_linear_NonDynamicallyQuantizableLinear': <class 'torch.nn.modules.linear.NonDynamicallyQuantizableLinear'>, 'torch_nn_modules_loss_BCELoss': <class 'torch.nn.modules.loss.BCELoss'>, 'torch_nn_modules_loss_BCEWithLogitsLoss': <class 'torch.nn.modules.loss.BCEWithLogitsLoss'>, 'torch_nn_modules_loss_CTCLoss': <class 'torch.nn.modules.loss.CTCLoss'>, 'torch_nn_modules_loss_CosineEmbeddingLoss': <class 'torch.nn.modules.loss.CosineEmbeddingLoss'>, 'torch_nn_modules_loss_CrossEntropyLoss': <class 'torch.nn.modules.loss.CrossEntropyLoss'>, 'torch_nn_modules_loss_GaussianNLLLoss': <class 'torch.nn.modules.loss.GaussianNLLLoss'>, 'torch_nn_modules_loss_HingeEmbeddingLoss': <class 'torch.nn.modules.loss.HingeEmbeddingLoss'>, 'torch_nn_modules_loss_HuberLoss': <class 'torch.nn.modules.loss.HuberLoss'>, 'torch_nn_modules_loss_KLDivLoss': <class 'torch.nn.modules.loss.KLDivLoss'>, 'torch_nn_modules_loss_L1Loss': <class 'torch.nn.modules.loss.L1Loss'>, 'torch_nn_modules_loss_MSELoss': <class 'torch.nn.modules.loss.MSELoss'>, 'torch_nn_modules_loss_MarginRankingLoss': <class 'torch.nn.modules.loss.MarginRankingLoss'>, 'torch_nn_modules_loss_MultiLabelMarginLoss': <class 'torch.nn.modules.loss.MultiLabelMarginLoss'>, 'torch_nn_modules_loss_MultiLabelSoftMarginLoss': <class 'torch.nn.modules.loss.MultiLabelSoftMarginLoss'>, 'torch_nn_modules_loss_MultiMarginLoss': <class 'torch.nn.modules.loss.MultiMarginLoss'>, 'torch_nn_modules_loss_NLLLoss': <class 'torch.nn.modules.loss.NLLLoss'>, 'torch_nn_modules_loss_NLLLoss2d': <class 'torch.nn.modules.loss.NLLLoss2d'>, 'torch_nn_modules_loss_PoissonNLLLoss': <class 'torch.nn.modules.loss.PoissonNLLLoss'>, 'torch_nn_modules_loss_SmoothL1Loss': <class 'torch.nn.modules.loss.SmoothL1Loss'>, 'torch_nn_modules_loss_SoftMarginLoss': <class 'torch.nn.modules.loss.SoftMarginLoss'>, 'torch_nn_modules_loss_TripletMarginLoss': <class 'torch.nn.modules.loss.TripletMarginLoss'>, 'torch_nn_modules_loss_TripletMarginWithDistanceLoss': <class 'torch.nn.modules.loss.TripletMarginWithDistanceLoss'>, 'torch_nn_modules_loss__Loss': <class 'torch.nn.modules.loss._Loss'>, 'torch_nn_modules_loss__WeightedLoss': <class 'torch.nn.modules.loss._WeightedLoss'>, 'torch_nn_modules_normalization_CrossMapLRN2d': <class 'torch.nn.modules.normalization.CrossMapLRN2d'>, 'torch_nn_modules_normalization_GroupNorm': <class 'torch.nn.modules.normalization.GroupNorm'>, 'torch_nn_modules_normalization_LayerNorm': <class 'torch.nn.modules.normalization.LayerNorm'>, 'torch_nn_modules_normalization_LocalResponseNorm': <class 'torch.nn.modules.normalization.LocalResponseNorm'>, 'torch_nn_modules_normalization_RMSNorm': <class 'torch.nn.modules.normalization.RMSNorm'>, 'torch_nn_modules_padding_CircularPad1d': <class 'torch.nn.modules.padding.CircularPad1d'>, 'torch_nn_modules_padding_CircularPad2d': <class 'torch.nn.modules.padding.CircularPad2d'>, 'torch_nn_modules_padding_CircularPad3d': <class 'torch.nn.modules.padding.CircularPad3d'>, 'torch_nn_modules_padding_ConstantPad1d': <class 'torch.nn.modules.padding.ConstantPad1d'>, 'torch_nn_modules_padding_ConstantPad2d': <class 'torch.nn.modules.padding.ConstantPad2d'>, 'torch_nn_modules_padding_ConstantPad3d': <class 'torch.nn.modules.padding.ConstantPad3d'>, 'torch_nn_modules_padding_ReflectionPad1d': <class 'torch.nn.modules.padding.ReflectionPad1d'>, 'torch_nn_modules_padding_ReflectionPad2d': <class 'torch.nn.modules.padding.ReflectionPad2d'>, 'torch_nn_modules_padding_ReflectionPad3d': <class 'torch.nn.modules.padding.ReflectionPad3d'>, 'torch_nn_modules_padding_ReplicationPad1d': <class 'torch.nn.modules.padding.ReplicationPad1d'>, 'torch_nn_modules_padding_ReplicationPad2d': <class 'torch.nn.modules.padding.ReplicationPad2d'>, 'torch_nn_modules_padding_ReplicationPad3d': <class 'torch.nn.modules.padding.ReplicationPad3d'>, 'torch_nn_modules_padding_ZeroPad1d': <class 'torch.nn.modules.padding.ZeroPad1d'>, 'torch_nn_modules_padding_ZeroPad2d': <class 'torch.nn.modules.padding.ZeroPad2d'>, 'torch_nn_modules_padding_ZeroPad3d': <class 'torch.nn.modules.padding.ZeroPad3d'>, 'torch_nn_modules_padding__CircularPadNd': <class 'torch.nn.modules.padding._CircularPadNd'>, 'torch_nn_modules_padding__ConstantPadNd': <class 'torch.nn.modules.padding._ConstantPadNd'>, 'torch_nn_modules_padding__ReflectionPadNd': <class 'torch.nn.modules.padding._ReflectionPadNd'>, 'torch_nn_modules_padding__ReplicationPadNd': <class 'torch.nn.modules.padding._ReplicationPadNd'>, 'torch_nn_modules_pixelshuffle_PixelShuffle': <class 'torch.nn.modules.pixelshuffle.PixelShuffle'>, 'torch_nn_modules_pixelshuffle_PixelUnshuffle': <class 'torch.nn.modules.pixelshuffle.PixelUnshuffle'>, 'torch_nn_modules_pooling_AdaptiveAvgPool1d': <class 'torch.nn.modules.pooling.AdaptiveAvgPool1d'>, 'torch_nn_modules_pooling_AdaptiveAvgPool2d': <class 'torch.nn.modules.pooling.AdaptiveAvgPool2d'>, 'torch_nn_modules_pooling_AdaptiveAvgPool3d': <class 'torch.nn.modules.pooling.AdaptiveAvgPool3d'>, 'torch_nn_modules_pooling_AdaptiveMaxPool1d': <class 'torch.nn.modules.pooling.AdaptiveMaxPool1d'>, 'torch_nn_modules_pooling_AdaptiveMaxPool2d': <class 'torch.nn.modules.pooling.AdaptiveMaxPool2d'>, 'torch_nn_modules_pooling_AdaptiveMaxPool3d': <class 'torch.nn.modules.pooling.AdaptiveMaxPool3d'>, 'torch_nn_modules_pooling_AvgPool1d': <class 'torch.nn.modules.pooling.AvgPool1d'>, 'torch_nn_modules_pooling_AvgPool2d': <class 'torch.nn.modules.pooling.AvgPool2d'>, 'torch_nn_modules_pooling_AvgPool3d': <class 'torch.nn.modules.pooling.AvgPool3d'>, 'torch_nn_modules_pooling_FractionalMaxPool2d': <class 'torch.nn.modules.pooling.FractionalMaxPool2d'>, 'torch_nn_modules_pooling_FractionalMaxPool3d': <class 'torch.nn.modules.pooling.FractionalMaxPool3d'>, 'torch_nn_modules_pooling_LPPool1d': <class 'torch.nn.modules.pooling.LPPool1d'>, 'torch_nn_modules_pooling_LPPool2d': <class 'torch.nn.modules.pooling.LPPool2d'>, 'torch_nn_modules_pooling_LPPool3d': <class 'torch.nn.modules.pooling.LPPool3d'>, 'torch_nn_modules_pooling_MaxPool1d': <class 'torch.nn.modules.pooling.MaxPool1d'>, 'torch_nn_modules_pooling_MaxPool2d': <class 'torch.nn.modules.pooling.MaxPool2d'>, 'torch_nn_modules_pooling_MaxPool3d': <class 'torch.nn.modules.pooling.MaxPool3d'>, 'torch_nn_modules_pooling_MaxUnpool1d': <class 'torch.nn.modules.pooling.MaxUnpool1d'>, 'torch_nn_modules_pooling_MaxUnpool2d': <class 'torch.nn.modules.pooling.MaxUnpool2d'>, 'torch_nn_modules_pooling_MaxUnpool3d': <class 'torch.nn.modules.pooling.MaxUnpool3d'>, 'torch_nn_modules_pooling__AdaptiveAvgPoolNd': <class 'torch.nn.modules.pooling._AdaptiveAvgPoolNd'>, 'torch_nn_modules_pooling__AdaptiveMaxPoolNd': <class 'torch.nn.modules.pooling._AdaptiveMaxPoolNd'>, 'torch_nn_modules_pooling__AvgPoolNd': <class 'torch.nn.modules.pooling._AvgPoolNd'>, 'torch_nn_modules_pooling__LPPoolNd': <class 'torch.nn.modules.pooling._LPPoolNd'>, 'torch_nn_modules_pooling__MaxPoolNd': <class 'torch.nn.modules.pooling._MaxPoolNd'>, 'torch_nn_modules_pooling__MaxUnpoolNd': <class 'torch.nn.modules.pooling._MaxUnpoolNd'>, 'torch_nn_modules_rnn_GRU': <class 'torch.nn.modules.rnn.GRU'>, 'torch_nn_modules_rnn_GRUCell': <class 'torch.nn.modules.rnn.GRUCell'>, 'torch_nn_modules_rnn_LSTM': <class 'torch.nn.modules.rnn.LSTM'>, 'torch_nn_modules_rnn_LSTMCell': <class 'torch.nn.modules.rnn.LSTMCell'>, 'torch_nn_modules_rnn_RNN': <class 'torch.nn.modules.rnn.RNN'>, 'torch_nn_modules_rnn_RNNBase': <class 'torch.nn.modules.rnn.RNNBase'>, 'torch_nn_modules_rnn_RNNCell': <class 'torch.nn.modules.rnn.RNNCell'>, 'torch_nn_modules_rnn_RNNCellBase': <class 'torch.nn.modules.rnn.RNNCellBase'>, 'torch_nn_modules_sparse_Embedding': <class 'torch.nn.modules.sparse.Embedding'>, 'torch_nn_modules_sparse_EmbeddingBag': <class 'torch.nn.modules.sparse.EmbeddingBag'>, 'torch_nn_modules_transformer_Transformer': <class 'torch.nn.modules.transformer.Transformer'>, 'torch_nn_modules_transformer_TransformerDecoder': <class 'torch.nn.modules.transformer.TransformerDecoder'>, 'torch_nn_modules_transformer_TransformerDecoderLayer': <class 'torch.nn.modules.transformer.TransformerDecoderLayer'>, 'torch_nn_modules_transformer_TransformerEncoder': <class 'torch.nn.modules.transformer.TransformerEncoder'>, 'torch_nn_modules_transformer_TransformerEncoderLayer': <class 'torch.nn.modules.transformer.TransformerEncoderLayer'>, 'torch_nn_modules_upsampling_Upsample': <class 'torch.nn.modules.upsampling.Upsample'>, 'torch_nn_modules_upsampling_UpsamplingBilinear2d': <class 'torch.nn.modules.upsampling.UpsamplingBilinear2d'>, 'torch_nn_modules_upsampling_UpsamplingNearest2d': <class 'torch.nn.modules.upsampling.UpsamplingNearest2d'>, 'torch_nn_parallel_DistributedDataParallelCPU': <class 'torch.nn.parallel.DistributedDataParallelCPU'>, 'torch_nn_parallel_data_parallel_DataParallel': <class 'torch.nn.parallel.data_parallel.DataParallel'>, 'torch_nn_parallel_distributed_DistributedDataParallel': <class 'torch.nn.parallel.distributed.DistributedDataParallel'>, 'torch_nn_utils_parametrizations__Orthogonal': <class 'torch.nn.utils.parametrizations._Orthogonal'>, 'torch_nn_utils_parametrizations__SpectralNorm': <class 'torch.nn.utils.parametrizations._SpectralNorm'>, 'torch_nn_utils_parametrizations__WeightNorm': <class 'torch.nn.utils.parametrizations._WeightNorm'>, 'torch_nn_utils_parametrize_ParametrizationList': <class 'torch.nn.utils.parametrize.ParametrizationList'>, 'torch_optim_swa_utils_AveragedModel': <class 'torch.optim.swa_utils.AveragedModel'>, 'torchaudio_models__hdemucs_HDemucs': <class 'torchaudio.models._hdemucs.HDemucs'>, 'torchaudio_models__hdemucs__BLSTM': <class 'torchaudio.models._hdemucs._BLSTM'>, 'torchaudio_models__hdemucs__DConv': <class 'torchaudio.models._hdemucs._DConv'>, 'torchaudio_models__hdemucs__HDecLayer': <class 'torchaudio.models._hdemucs._HDecLayer'>, 'torchaudio_models__hdemucs__HEncLayer': <class 'torchaudio.models._hdemucs._HEncLayer'>, 'torchaudio_models__hdemucs__LayerScale': <class 'torchaudio.models._hdemucs._LayerScale'>, 'torchaudio_models__hdemucs__LocalState': <class 'torchaudio.models._hdemucs._LocalState'>, 'torchaudio_models__hdemucs__ScaledEmbedding': <class 'torchaudio.models._hdemucs._ScaledEmbedding'>, 'torchaudio_models_conformer_Conformer': <class 'torchaudio.models.conformer.Conformer'>, 'torchaudio_models_conformer_ConformerLayer': <class 'torchaudio.models.conformer.ConformerLayer'>, 'torchaudio_models_conformer__ConvolutionModule': <class 'torchaudio.models.conformer._ConvolutionModule'>, 'torchaudio_models_conformer__FeedForwardModule': <class 'torchaudio.models.conformer._FeedForwardModule'>, 'torchaudio_models_conv_tasnet_ConvBlock': <class 'torchaudio.models.conv_tasnet.ConvBlock'>, 'torchaudio_models_conv_tasnet_ConvTasNet': <class 'torchaudio.models.conv_tasnet.ConvTasNet'>, 'torchaudio_models_conv_tasnet_MaskGenerator': <class 'torchaudio.models.conv_tasnet.MaskGenerator'>, 'torchaudio_models_deepspeech_DeepSpeech': <class 'torchaudio.models.deepspeech.DeepSpeech'>, 'torchaudio_models_deepspeech_FullyConnected': <class 'torchaudio.models.deepspeech.FullyConnected'>, 'torchaudio_models_emformer_Emformer': <class 'torchaudio.models.emformer.Emformer'>, 'torchaudio_models_emformer__EmformerAttention': <class 'torchaudio.models.emformer._EmformerAttention'>, 'torchaudio_models_emformer__EmformerImpl': <class 'torchaudio.models.emformer._EmformerImpl'>, 'torchaudio_models_emformer__EmformerLayer': <class 'torchaudio.models.emformer._EmformerLayer'>, 'torchaudio_models_rnnt_RNNT': <class 'torchaudio.models.rnnt.RNNT'>, 'torchaudio_models_rnnt__CustomLSTM': <class 'torchaudio.models.rnnt._CustomLSTM'>, 'torchaudio_models_rnnt__EmformerEncoder': <class 'torchaudio.models.rnnt._EmformerEncoder'>, 'torchaudio_models_rnnt__Joiner': <class 'torchaudio.models.rnnt._Joiner'>, 'torchaudio_models_rnnt__Predictor': <class 'torchaudio.models.rnnt._Predictor'>, 'torchaudio_models_rnnt__TimeReduction': <class 'torchaudio.models.rnnt._TimeReduction'>, 'torchaudio_models_rnnt_decoder_RNNTBeamSearch': <class 'torchaudio.models.rnnt_decoder.RNNTBeamSearch'>, 'torchaudio_models_squim_objective_AutoPool': <class 'torchaudio.models.squim.objective.AutoPool'>, 'torchaudio_models_squim_objective_DPRNN': <class 'torchaudio.models.squim.objective.DPRNN'>, 'torchaudio_models_squim_objective_Encoder': <class 'torchaudio.models.squim.objective.Encoder'>, 'torchaudio_models_squim_objective_RangeSigmoid': <class 'torchaudio.models.squim.objective.RangeSigmoid'>, 'torchaudio_models_squim_objective_SingleRNN': <class 'torchaudio.models.squim.objective.SingleRNN'>, 'torchaudio_models_squim_objective_SquimObjective': <class 'torchaudio.models.squim.objective.SquimObjective'>, 'torchaudio_models_squim_subjective_AttPool': <class 'torchaudio.models.squim.subjective.AttPool'>, 'torchaudio_models_squim_subjective_Predictor': <class 'torchaudio.models.squim.subjective.Predictor'>, 'torchaudio_models_squim_subjective_SquimSubjective': <class 'torchaudio.models.squim.subjective.SquimSubjective'>, 'torchaudio_models_tacotron2_Tacotron2': <class 'torchaudio.models.tacotron2.Tacotron2'>, 'torchaudio_models_tacotron2__Attention': <class 'torchaudio.models.tacotron2._Attention'>, 'torchaudio_models_tacotron2__Decoder': <class 'torchaudio.models.tacotron2._Decoder'>, 'torchaudio_models_tacotron2__Encoder': <class 'torchaudio.models.tacotron2._Encoder'>, 'torchaudio_models_tacotron2__LocationLayer': <class 'torchaudio.models.tacotron2._LocationLayer'>, 'torchaudio_models_tacotron2__Postnet': <class 'torchaudio.models.tacotron2._Postnet'>, 'torchaudio_models_tacotron2__Prenet': <class 'torchaudio.models.tacotron2._Prenet'>, 'torchaudio_models_wav2letter_Wav2Letter': <class 'torchaudio.models.wav2letter.Wav2Letter'>, 'torchaudio_models_wav2vec2_components_ConvLayerBlock': <class 'torchaudio.models.wav2vec2.components.ConvLayerBlock'>, 'torchaudio_models_wav2vec2_components_ConvolutionalPositionalEmbedding': <class 'torchaudio.models.wav2vec2.components.ConvolutionalPositionalEmbedding'>, 'torchaudio_models_wav2vec2_components_Encoder': <class 'torchaudio.models.wav2vec2.components.Encoder'>, 'torchaudio_models_wav2vec2_components_EncoderLayer': <class 'torchaudio.models.wav2vec2.components.EncoderLayer'>, 'torchaudio_models_wav2vec2_components_FeatureExtractor': <class 'torchaudio.models.wav2vec2.components.FeatureExtractor'>, 'torchaudio_models_wav2vec2_components_FeatureProjection': <class 'torchaudio.models.wav2vec2.components.FeatureProjection'>, 'torchaudio_models_wav2vec2_components_FeedForward': <class 'torchaudio.models.wav2vec2.components.FeedForward'>, 'torchaudio_models_wav2vec2_components_LayerNorm': <class 'torchaudio.models.wav2vec2.components.LayerNorm'>, 'torchaudio_models_wav2vec2_components_LogitGenerator': <class 'torchaudio.models.wav2vec2.components.LogitGenerator'>, 'torchaudio_models_wav2vec2_components_MaskGenerator': <class 'torchaudio.models.wav2vec2.components.MaskGenerator'>, 'torchaudio_models_wav2vec2_components_SelfAttention': <class 'torchaudio.models.wav2vec2.components.SelfAttention'>, 'torchaudio_models_wav2vec2_components_Transformer': <class 'torchaudio.models.wav2vec2.components.Transformer'>, 'torchaudio_models_wav2vec2_model_HuBERTPretrainModel': <class 'torchaudio.models.wav2vec2.model.HuBERTPretrainModel'>, 'torchaudio_models_wav2vec2_model_Wav2Vec2Model': <class 'torchaudio.models.wav2vec2.model.Wav2Vec2Model'>, 'torchaudio_models_wav2vec2_wavlm_attention_WavLMSelfAttention': <class 'torchaudio.models.wav2vec2.wavlm_attention.WavLMSelfAttention'>, 'torchaudio_models_wavernn_MelResNet': <class 'torchaudio.models.wavernn.MelResNet'>, 'torchaudio_models_wavernn_ResBlock': <class 'torchaudio.models.wavernn.ResBlock'>, 'torchaudio_models_wavernn_Stretch2d': <class 'torchaudio.models.wavernn.Stretch2d'>, 'torchaudio_models_wavernn_UpsampleNetwork': <class 'torchaudio.models.wavernn.UpsampleNetwork'>, 'torchaudio_models_wavernn_WaveRNN': <class 'torchaudio.models.wavernn.WaveRNN'>, 'torchaudio_pipelines__tts_impl__GriffinLimVocoder': <class 'torchaudio.pipelines._tts.impl._GriffinLimVocoder'>, 'torchaudio_pipelines__tts_impl__WaveRNNVocoder': <class 'torchaudio.pipelines._tts.impl._WaveRNNVocoder'>, 'torchaudio_pipelines__wav2vec2_utils__Wav2Vec2Model': <class 'torchaudio.pipelines._wav2vec2.utils._Wav2Vec2Model'>, 'torchaudio_pipelines_rnnt_pipeline__FunctionalModule': <class 'torchaudio.pipelines.rnnt_pipeline._FunctionalModule'>, 'torchaudio_pipelines_rnnt_pipeline__GlobalStatsNormalization': <class 'torchaudio.pipelines.rnnt_pipeline._GlobalStatsNormalization'>, 'torchaudio_pipelines_rnnt_pipeline__ModuleFeatureExtractor': <class 'torchaudio.pipelines.rnnt_pipeline._ModuleFeatureExtractor'>, 'torchaudio_transforms__multi_channel_MVDR': <class 'torchaudio.transforms._multi_channel.MVDR'>, 'torchaudio_transforms__multi_channel_PSD': <class 'torchaudio.transforms._multi_channel.PSD'>, 'torchaudio_transforms__multi_channel_RTFMVDR': <class 'torchaudio.transforms._multi_channel.RTFMVDR'>, 'torchaudio_transforms__multi_channel_SoudenMVDR': <class 'torchaudio.transforms._multi_channel.SoudenMVDR'>, 'torchaudio_transforms__transforms_AddNoise': <class 'torchaudio.transforms._transforms.AddNoise'>, 'torchaudio_transforms__transforms_AmplitudeToDB': <class 'torchaudio.transforms._transforms.AmplitudeToDB'>, 'torchaudio_transforms__transforms_ComputeDeltas': <class 'torchaudio.transforms._transforms.ComputeDeltas'>, 'torchaudio_transforms__transforms_Convolve': <class 'torchaudio.transforms._transforms.Convolve'>, 'torchaudio_transforms__transforms_Deemphasis': <class 'torchaudio.transforms._transforms.Deemphasis'>, 'torchaudio_transforms__transforms_FFTConvolve': <class 'torchaudio.transforms._transforms.FFTConvolve'>, 'torchaudio_transforms__transforms_Fade': <class 'torchaudio.transforms._transforms.Fade'>, 'torchaudio_transforms__transforms_FrequencyMasking': <class 'torchaudio.transforms._transforms.FrequencyMasking'>, 'torchaudio_transforms__transforms_GriffinLim': <class 'torchaudio.transforms._transforms.GriffinLim'>, 'torchaudio_transforms__transforms_InverseMelScale': <class 'torchaudio.transforms._transforms.InverseMelScale'>, 'torchaudio_transforms__transforms_InverseSpectrogram': <class 'torchaudio.transforms._transforms.InverseSpectrogram'>, 'torchaudio_transforms__transforms_LFCC': <class 'torchaudio.transforms._transforms.LFCC'>, 'torchaudio_transforms__transforms_Loudness': <class 'torchaudio.transforms._transforms.Loudness'>, 'torchaudio_transforms__transforms_MFCC': <class 'torchaudio.transforms._transforms.MFCC'>, 'torchaudio_transforms__transforms_MelScale': <class 'torchaudio.transforms._transforms.MelScale'>, 'torchaudio_transforms__transforms_MelSpectrogram': <class 'torchaudio.transforms._transforms.MelSpectrogram'>, 'torchaudio_transforms__transforms_MuLawDecoding': <class 'torchaudio.transforms._transforms.MuLawDecoding'>, 'torchaudio_transforms__transforms_MuLawEncoding': <class 'torchaudio.transforms._transforms.MuLawEncoding'>, 'torchaudio_transforms__transforms_PitchShift': <class 'torchaudio.transforms._transforms.PitchShift'>, 'torchaudio_transforms__transforms_Preemphasis': <class 'torchaudio.transforms._transforms.Preemphasis'>, 'torchaudio_transforms__transforms_RNNTLoss': <class 'torchaudio.transforms._transforms.RNNTLoss'>, 'torchaudio_transforms__transforms_Resample': <class 'torchaudio.transforms._transforms.Resample'>, 'torchaudio_transforms__transforms_SlidingWindowCmn': <class 'torchaudio.transforms._transforms.SlidingWindowCmn'>, 'torchaudio_transforms__transforms_SpecAugment': <class 'torchaudio.transforms._transforms.SpecAugment'>, 'torchaudio_transforms__transforms_SpectralCentroid': <class 'torchaudio.transforms._transforms.SpectralCentroid'>, 'torchaudio_transforms__transforms_Spectrogram': <class 'torchaudio.transforms._transforms.Spectrogram'>, 'torchaudio_transforms__transforms_Speed': <class 'torchaudio.transforms._transforms.Speed'>, 'torchaudio_transforms__transforms_SpeedPerturbation': <class 'torchaudio.transforms._transforms.SpeedPerturbation'>, 'torchaudio_transforms__transforms_TimeMasking': <class 'torchaudio.transforms._transforms.TimeMasking'>, 'torchaudio_transforms__transforms_TimeStretch': <class 'torchaudio.transforms._transforms.TimeStretch'>, 'torchaudio_transforms__transforms_Vad': <class 'torchaudio.transforms._transforms.Vad'>, 'torchaudio_transforms__transforms_Vol': <class 'torchaudio.transforms._transforms.Vol'>, 'torchaudio_transforms__transforms__AxisMasking': <class 'torchaudio.transforms._transforms._AxisMasking'>}

classmethod init_weights(model, config, prefix='')[source]

Parameters:

model (Module)
config (Config | None)
prefix (str)

Return type:

None

class todd.registries.OptimizerRegistry[source]

Bases: Registry

data = {'ASGD': <class 'torch.optim.asgd.ASGD'>, 'Adadelta': <class 'torch.optim.adadelta.Adadelta'>, 'Adafactor': <class 'torch.optim.Adafactor'>, 'Adagrad': <class 'torch.optim.adagrad.Adagrad'>, 'Adam': <class 'torch.optim.adam.Adam'>, 'AdamW': <class 'torch.optim.adamw.AdamW'>, 'Adamax': <class 'torch.optim.adamax.Adamax'>, 'LBFGS': <class 'torch.optim.lbfgs.LBFGS'>, 'Muon': <class 'torch.optim.Muon'>, 'NAdam': <class 'torch.optim.nadam.NAdam'>, 'RAdam': <class 'torch.optim.radam.RAdam'>, 'RMSprop': <class 'torch.optim.rmsprop.RMSprop'>, 'Rprop': <class 'torch.optim.rprop.Rprop'>, 'SGD': <class 'torch.optim.sgd.SGD'>, 'SparseAdam': <class 'torch.optim.sparse_adam.SparseAdam'>}

static params(model, config)[source]

Parameters:

model (Module)
config (Config)

Return type:

Config

class todd.registries.PartialRegistry[source]

Bases: object

data = {}

class todd.registries.PartialRegistryMeta[source]: Bases: RegistryMeta

class todd.registries.RunnerRegistry[source]

Bases: Registry

data = {'BaseRunner': <class 'todd.runners.base.BaseRunner'>, 'EpochBasedTrainer': <class 'todd.runners.epoch_based_trainer.EpochBasedTrainer'>, 'IterBasedTrainer': <class 'todd.runners.iter_based_trainer.IterBasedTrainer'>, 'Trainer': <class 'todd.runners.trainer.Trainer'>, 'Validator': <class 'todd.runners.validator.Validator'>}

class todd.registries.SamplerRegistry[source]

Bases: Registry

data = {'BatchSampler': <class 'torch.utils.data.sampler.BatchSampler'>, 'DistributedSampler': <class 'torch.utils.data.distributed.DistributedSampler'>, 'RandomSampler': <class 'torch.utils.data.sampler.RandomSampler'>, 'SequentialSampler': <class 'torch.utils.data.sampler.SequentialSampler'>, 'SubsetRandomSampler': <class 'torch.utils.data.sampler.SubsetRandomSampler'>, 'WeightedRandomSampler': <class 'torch.utils.data.sampler.WeightedRandomSampler'>, '_InfiniteConstantSampler': <class 'torch.utils.data.dataloader._InfiniteConstantSampler'>}

class todd.registries.TaskRegistry[source]

Bases: Registry

data = {}

class todd.registries.TransformRegistry[source]

Bases: Registry

data = {'AugMix': <class 'torchvision.transforms.v2._auto_augment.AugMix'>, 'AutoAugment': <class 'torchvision.transforms.v2._auto_augment.AutoAugment'>, 'AutoAugmentPolicy': <enum 'AutoAugmentPolicy'>, 'CenterCrop': <class 'torchvision.transforms.v2._geometry.CenterCrop'>, 'ClampBoundingBoxes': <class 'torchvision.transforms.v2._meta.ClampBoundingBoxes'>, 'ClampKeyPoints': <class 'torchvision.transforms.v2._meta.ClampKeyPoints'>, 'ColorJitter': <class 'torchvision.transforms.v2._color.ColorJitter'>, 'Compose': <class 'torchvision.transforms.v2._container.Compose'>, 'ConvertBoundingBoxFormat': <class 'torchvision.transforms.v2._meta.ConvertBoundingBoxFormat'>, 'ConvertImageDtype': <class 'torchvision.transforms.v2._misc.ConvertImageDtype'>, 'CutMix': <class 'torchvision.transforms.v2._augment.CutMix'>, 'ElasticTransform': <class 'torchvision.transforms.v2._geometry.ElasticTransform'>, 'FiveCrop': <class 'torchvision.transforms.v2._geometry.FiveCrop'>, 'GaussianBlur': <class 'torchvision.transforms.v2._misc.GaussianBlur'>, 'GaussianNoise': <class 'torchvision.transforms.v2._misc.GaussianNoise'>, 'Grayscale': <class 'torchvision.transforms.v2._color.Grayscale'>, 'Identity': <class 'torchvision.transforms.v2._misc.Identity'>, 'InterpolationMode': <enum 'InterpolationMode'>, 'JPEG': <class 'torchvision.transforms.v2._augment.JPEG'>, 'Lambda': <class 'torchvision.transforms.v2._misc.Lambda'>, 'LinearTransformation': <class 'torchvision.transforms.v2._misc.LinearTransformation'>, 'MixUp': <class 'torchvision.transforms.v2._augment.MixUp'>, 'Normalize': <class 'torchvision.transforms.v2._misc.Normalize'>, 'PILToTensor': <class 'torchvision.transforms.v2._type_conversion.PILToTensor'>, 'Pad': <class 'torchvision.transforms.v2._geometry.Pad'>, 'RGB': <class 'torchvision.transforms.v2._color.RGB'>, 'RandAugment': <class 'torchvision.transforms.v2._auto_augment.RandAugment'>, 'RandomAdjustSharpness': <class 'torchvision.transforms.v2._color.RandomAdjustSharpness'>, 'RandomAffine': <class 'torchvision.transforms.v2._geometry.RandomAffine'>, 'RandomApply': <class 'torchvision.transforms.v2._container.RandomApply'>, 'RandomAutocontrast': <class 'torchvision.transforms.v2._color.RandomAutocontrast'>, 'RandomChannelPermutation': <class 'torchvision.transforms.v2._color.RandomChannelPermutation'>, 'RandomChoice': <class 'torchvision.transforms.v2._container.RandomChoice'>, 'RandomCrop': <class 'torchvision.transforms.v2._geometry.RandomCrop'>, 'RandomEqualize': <class 'torchvision.transforms.v2._color.RandomEqualize'>, 'RandomErasing': <class 'torchvision.transforms.v2._augment.RandomErasing'>, 'RandomGrayscale': <class 'torchvision.transforms.v2._color.RandomGrayscale'>, 'RandomHorizontalFlip': <class 'torchvision.transforms.v2._geometry.RandomHorizontalFlip'>, 'RandomInvert': <class 'torchvision.transforms.v2._color.RandomInvert'>, 'RandomIoUCrop': <class 'torchvision.transforms.v2._geometry.RandomIoUCrop'>, 'RandomOrder': <class 'torchvision.transforms.v2._container.RandomOrder'>, 'RandomPerspective': <class 'torchvision.transforms.v2._geometry.RandomPerspective'>, 'RandomPhotometricDistort': <class 'torchvision.transforms.v2._color.RandomPhotometricDistort'>, 'RandomPosterize': <class 'torchvision.transforms.v2._color.RandomPosterize'>, 'RandomResize': <class 'torchvision.transforms.v2._geometry.RandomResize'>, 'RandomResizedCrop': <class 'torchvision.transforms.v2._geometry.RandomResizedCrop'>, 'RandomRotation': <class 'torchvision.transforms.v2._geometry.RandomRotation'>, 'RandomShortestSize': <class 'torchvision.transforms.v2._geometry.RandomShortestSize'>, 'RandomSolarize': <class 'torchvision.transforms.v2._color.RandomSolarize'>, 'RandomVerticalFlip': <class 'torchvision.transforms.v2._geometry.RandomVerticalFlip'>, 'RandomZoomOut': <class 'torchvision.transforms.v2._geometry.RandomZoomOut'>, 'Resize': <class 'torchvision.transforms.v2._geometry.Resize'>, 'SanitizeBoundingBoxes': <class 'torchvision.transforms.v2._misc.SanitizeBoundingBoxes'>, 'SanitizeKeyPoints': <class 'torchvision.transforms.v2._misc.SanitizeKeyPoints'>, 'ScaleJitter': <class 'torchvision.transforms.v2._geometry.ScaleJitter'>, 'SetClampingMode': <class 'torchvision.transforms.v2._meta.SetClampingMode'>, 'TenCrop': <class 'torchvision.transforms.v2._geometry.TenCrop'>, 'ToDtype': <class 'torchvision.transforms.v2._misc.ToDtype'>, 'ToImage': <class 'torchvision.transforms.v2._type_conversion.ToImage'>, 'ToPILImage': <class 'torchvision.transforms.v2._type_conversion.ToPILImage'>, 'ToPureTensor': <class 'torchvision.transforms.v2._type_conversion.ToPureTensor'>, 'ToTensor': <class 'torchvision.transforms.v2._deprecated.ToTensor'>, 'Transform': <class 'torchvision.transforms.v2._transform.Transform'>, 'TrivialAugmentWide': <class 'torchvision.transforms.v2._auto_augment.TrivialAugmentWide'>, 'UniformTemporalSubsample': <class 'torchvision.transforms.v2._temporal.UniformTemporalSubsample'>}

class todd.registries.VisualRegistry[source]

Bases: Registry

data = {'CV2Visual': <class 'todd.visuals.cv2.CV2Visual'>, 'PPTXVisual': <class 'todd.visuals.pptx.PPTXVisual'>}

class todd.registries.WorkerInitRegistry[source]

Bases: PartialRegistry

data = {'default': <function default_worker_init>}