Fix example script

2021-05-18 10:15:38 +02:00 · 2021-05-18 10:15:38 +02:00 · 00cdacf7ae
commit 00cdacf7ae
parent 4957e821f6
1 changed files with 1 additions and 1 deletions
--- a/examples/glvq_spiral.py
+++ b/examples/glvq_spiral.py
@ -30,7 +30,7 @@ if __name__ == "__main__":
                                                                   noise=1e-1))

    # 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)

    # Setup trainer