litmodule

MNISTClassificationLitModule.

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

Bases: BaseClassificationLitModule

project BaseClassificationLitModule.

property wandb_media_x

Converts a tensor to a W&B media object.