py:class torchvision.transforms.v2._transform.Transform
py:class torch.nn.modules.loss._Loss
py:class Module
py:class optional
py:class torchvision.models._api.WeightsEnum
py:class click.formatting.HelpFormatter
