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
py:class tensorboardX.SummaryWriter
py:class pathlib._local.Path
py:class pydantic.types.StringConstraints
