Examples use GPUs if available.

2021-05-13 15:22:01 +02:00
parent 8f9c29bd2b
commit 0eac2ce326
14 changed files with 56 additions and 39 deletions
--- a/prototorch/models/glvq.py
+++ b/prototorch/models/glvq.py
@@ -84,7 +84,7 @@ class GLVQ(AbstractPrototypeModel):
            d = self(x)
            plabels = self.proto_layer.component_labels
            y_pred = wtac(d, plabels)
-        return y_pred.numpy()
+        return y_pred


 class LVQ1(GLVQ):