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