amanmibra commited on
Commit
ff57d28
·
1 Parent(s): b1f510e

Update print string

Browse files
Files changed (1) hide show
  1. train.py +1 -1
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 feed forward net saved at {model_filename}")
 
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}")