fix: style fixes
This commit is contained in:
@@ -72,4 +72,4 @@ if __name__ == "__main__":
|
||||
# Training loop
|
||||
trainer.fit(model, train_loader)
|
||||
|
||||
torch.save(model, "iris.pth")
|
||||
torch.save(model, "iris.pth")
|
||||
|
@@ -71,4 +71,4 @@ if __name__ == "__main__":
|
||||
# Training loop
|
||||
trainer.fit(model, train_loader)
|
||||
|
||||
torch.save(model, "iris.pth")
|
||||
torch.save(model, "iris.pth")
|
||||
|
Reference in New Issue
Block a user