fix: style fixes

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

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