All modules for which code is available
- pugh_torch
- pugh_torch.augmentations.transforms.resize_shortest
- pugh_torch.callbacks.histogram
- pugh_torch.callbacks.model_checkpoint
- pugh_torch.callbacks.tensorboard_add_classification
- pugh_torch.callbacks.tensorboard_add_depth
- pugh_torch.callbacks.tensorboard_add_ss
- pugh_torch.callbacks.tensorboard_base
- pugh_torch.datasets
- pugh_torch.datasets.base
- pugh_torch.datasets.classification.cifar
- pugh_torch.datasets.classification.imagenet
- pugh_torch.datasets.classification.mnist
- pugh_torch.datasets.nyuv2
- pugh_torch.datasets.segmentation.ade20k
- pugh_torch.datasets.torchvision
- pugh_torch.exceptions
- pugh_torch.helpers
- pugh_torch.linalg.batch_lstsq
- pugh_torch.losses
- pugh_torch.mappings.color
- pugh_torch.models.io
- pugh_torch.models.resnet
- pugh_torch.modules.activation
- pugh_torch.modules.conv
- pugh_torch.modules.hash
- pugh_torch.modules.init
- pugh_torch.modules.lightning_module
- pugh_torch.modules.load_state_dict_mixin
- pugh_torch.modules.meta
- pugh_torch.optimizers
- pugh_torch.optimizers.base
- pugh_torch.optimizers.lookahead
- pugh_torch.optimizers.ralamb
- pugh_torch.optimizers.rangerlars
- pugh_torch.optimizers.schedulers
- pugh_torch.transforms.imagenet
- pugh_torch.utils.batch_index_select
- pugh_torch.utils.io
- pugh_torch.utils.misc
- pugh_torch.utils.tensorboard
- torchvision.datasets.cifar
- torchvision.datasets.imagenet
- torchvision.datasets.mnist