prototorch_models/examples/cli/gmlvq.py

"""GLVQ example using the MNIST dataset."""

from prototorch.models import ImageGLVQ
from prototorch.models.data import train_on_mnist
from pytorch_lightning.utilities.cli import LightningCLI


class GLVQMNIST(train_on_mnist(batch_size=64), ImageGLVQ):
    """Model Definition."""


cli = LightningCLI(GLVQMNIST)
Remove normalization transform from cli example 2021-05-25 19:13:37 +00:00			`"""GLVQ example using the MNIST dataset."""`
LightningCLI Example. 2021-05-21 15:10:36 +00:00
			`from prototorch.models import ImageGLVQ`
Add MNIST datamodule and training mixin factory. 2021-05-28 14:33:31 +00:00			`from prototorch.models.data import train_on_mnist`
LightningCLI Example. 2021-05-21 15:10:36 +00:00			`from pytorch_lightning.utilities.cli import LightningCLI`


Add MNIST datamodule and training mixin factory. 2021-05-28 14:33:31 +00:00			`class GLVQMNIST(train_on_mnist(batch_size=64), ImageGLVQ):`
Remove normalization transform from cli example 2021-05-25 19:13:37 +00:00			`"""Model Definition."""`
LightningCLI Example. 2021-05-21 15:10:36 +00:00

Remove normalization transform from cli example 2021-05-25 19:13:37 +00:00			`cli = LightningCLI(GLVQMNIST)`