Small fix on example script
This commit is contained in:
parent
a16bebd0c4
commit
8bad54fc2d
@ -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(
|
||||
|
Loading…
Reference in New Issue
Block a user