From 16f410e80909ceaa6c362d3b76461748f6b0d61b Mon Sep 17 00:00:00 2001 From: Alexander Engelsberger Date: Thu, 9 Mar 2023 15:59:49 +0100 Subject: [PATCH] fix: style fixes --- examples/gmlvq_iris.py | 2 +- examples/grlvq_iris.py | 2 +- prototorch/models/glvq.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/examples/gmlvq_iris.py b/examples/gmlvq_iris.py index dee0a84..b713983 100644 --- a/examples/gmlvq_iris.py +++ b/examples/gmlvq_iris.py @@ -72,4 +72,4 @@ if __name__ == "__main__": # Training loop trainer.fit(model, train_loader) - torch.save(model, "iris.pth") \ No newline at end of file + torch.save(model, "iris.pth") diff --git a/examples/grlvq_iris.py b/examples/grlvq_iris.py index 2ede559..97e0a0c 100644 --- a/examples/grlvq_iris.py +++ b/examples/grlvq_iris.py @@ -71,4 +71,4 @@ if __name__ == "__main__": # Training loop trainer.fit(model, train_loader) - torch.save(model, "iris.pth") \ No newline at end of file + torch.save(model, "iris.pth") diff --git a/prototorch/models/glvq.py b/prototorch/models/glvq.py index d376c9a..2ca9834 100644 --- a/prototorch/models/glvq.py +++ b/prototorch/models/glvq.py @@ -274,7 +274,7 @@ class GMLVQ(GLVQ): omega = omega_initializer.generate(self.hparams["input_dim"], self.hparams["latent_dim"]) self.register_parameter("_omega", Parameter(omega)) - + @property def omega_matrix(self): return self._omega.detach().cpu()