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

fix: style fixes

Browse Source
This commit is contained in:
Alexander Engelsberger
2023-03-09 15:59:49 +01:00
parent 46dfb82371
commit 16f410e809
3 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/gmlvq_iris.py
View File

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

2
examples/grlvq_iris.py
View File

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

2
prototorch/models/glvq.py
View File

@@ -274,7 +274,7 @@ class GMLVQ(GLVQ):
omega = omega_initializer.generate(self.hparams["input_dim"], omega = omega_initializer.generate(self.hparams["input_dim"],
self.hparams["latent_dim"]) self.hparams["latent_dim"])
self.register_parameter("_omega", Parameter(omega)) self.register_parameter("_omega", Parameter(omega))
@property @property
def omega_matrix(self): def omega_matrix(self):
return self._omega.detach().cpu() return self._omega.detach().cpu()