py:class optional
py:class torch.nn.modules.loss._Loss
py:class tensorboardX.SummaryWriter
py:class pydantic.types.StringConstraints
