14 lines
278 B
Python
14 lines
278 B
Python
"""GLVQ example using the MNIST dataset."""
|
|
|
|
from prototorch.models import ImageGLVQ
|
|
from pytorch_lightning.utilities.cli import LightningCLI
|
|
|
|
from mnist import TrainOnMNIST
|
|
|
|
|
|
class GLVQMNIST(TrainOnMNIST, ImageGLVQ):
|
|
"""Model Definition."""
|
|
|
|
|
|
cli = LightningCLI(GLVQMNIST)
|