2021-05-25 19:13:37 +00:00
|
|
|
"""GLVQ example using the MNIST dataset."""
|
2021-05-21 15:10:36 +00:00
|
|
|
|
|
|
|
|
from prototorch.models import ImageGLVQ
|
|
|
|
|
from pytorch_lightning.utilities.cli import LightningCLI
|
|
|
|
|
|
|
|
|
|
from mnist import TrainOnMNIST
|
|
|
|
|
|
|
|
|
|
|
2021-05-25 19:13:37 +00:00
|
|
|
class GLVQMNIST(TrainOnMNIST, ImageGLVQ):
|
|
|
|
|
"""Model Definition."""
|
2021-05-21 15:10:36 +00:00
|
|
|
|
|
|
|
|
|
2021-05-25 19:13:37 +00:00
|
|
|
cli = LightningCLI(GLVQMNIST)
|
