Spaces:
Runtime error
Runtime error
Update print string
Browse files
train.py
CHANGED
@@ -79,4 +79,4 @@ if __name__ == "__main__":
|
|
79 |
now = now.strftime("%Y%m%d_%H%M%S")
|
80 |
model_filename = f"models/void_{now}.pth"
|
81 |
torch.save(model.state_dict(), model_filename)
|
82 |
-
print(f"Trained
|
|
|
79 |
now = now.strftime("%Y%m%d_%H%M%S")
|
80 |
model_filename = f"models/void_{now}.pth"
|
81 |
torch.save(model.state_dict(), model_filename)
|
82 |
+
print(f"Trained void model saved at {model_filename}")
|