litmodule

class projects.classify_mnist.litmodule.MNISTClassificationLitModule(*args, **kwargs)

Bases: BaseClassificationLitModule

property wandb_media_x

Converts a tensor to a W&B media object.