fix: style fixes

This commit is contained in:
Alexander Engelsberger 2023-03-09 15:59:49 +01:00
parent 46dfb82371
commit 16f410e809
No known key found for this signature in database
3 changed files with 3 additions and 3 deletions

View File

@ -72,4 +72,4 @@ if __name__ == "__main__":
# Training loop
trainer.fit(model, train_loader)
torch.save(model, "iris.pth")
torch.save(model, "iris.pth")

View File

@ -71,4 +71,4 @@ if __name__ == "__main__":
# Training loop
trainer.fit(model, train_loader)
torch.save(model, "iris.pth")
torch.save(model, "iris.pth")

View File

@ -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()