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

Fix example script

Browse Source
This commit is contained in:
Jensun Ravichandran 2021-05-18 10:15:38 +02:00
parent 4957e821f6
commit 00cdacf7ae
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_spiral.py
View File

@ -30,7 +30,7 @@ if __name__ == "__main__":
noise=1e-1)) noise=1e-1))
# Callbacks # Callbacks
vis = pt.models.VisGLVQ2D(train_ds, show_last_only=True) vis = pt.models.VisGLVQ2D(train_ds, show_last_only=True, block=True)
snan = StopOnNaN(model.proto_layer.components) snan = StopOnNaN(model.proto_layer.components)
# Setup trainer # Setup trainer