Contents:
pugh_torch
pugh_torch.augmentations
pugh_torch.augmentations.transforms
pugh_torch.augmentations.transforms.resize_shortest
pugh_torch.callbacks
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
pugh_torch.datasets.classification.cifar
pugh_torch.datasets.classification.imagenet
pugh_torch.datasets.classification.mnist
pugh_torch.datasets.nyuv2
pugh_torch.datasets.segmentation
pugh_torch.datasets.segmentation.ade20k
pugh_torch.datasets.torchvision
pugh_torch.exceptions
pugh_torch.helpers
pugh_torch.linalg
pugh_torch.linalg.batch_lstsq
pugh_torch.losses
pugh_torch.losses.hetero_cross_entropy
pugh_torch.losses.label_smoothing
pugh_torch.mappings
pugh_torch.mappings.color
pugh_torch.metrics
pugh_torch.models
pugh_torch.models.io
pugh_torch.models.resnet
pugh_torch.modules
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
pugh_torch.transforms.imagenet
pugh_torch.utils
pugh_torch.utils.batch_index_select
pugh_torch.utils.io
pugh_torch.utils.misc
pugh_torch.utils.tensorboard