Small fix on example script

This commit is contained in:
Jensun Ravichandran 2021-04-29 17:11:06 +02:00
parent a16bebd0c4
commit 8bad54fc2d

View File

@ -28,7 +28,7 @@ if __name__ == "__main__":
)
# Initialize the model
model = GLVQ(hparams, data=[x_train, y_train])
model = GLVQ(hparams)
# Setup trainer
trainer = pl.Trainer(