julius/prototorch_models
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Small fix on example script

Browse Source
This commit is contained in:
Jensun Ravichandran 2021-04-29 17:11:06 +02:00
parent a16bebd0c4
commit 8bad54fc2d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/glvq_iris_v1.py
View File

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