From d8cabe75de15811fe2ecc4b8c30d642669c37298 Mon Sep 17 00:00:00 2001
From: moghadas76 <s.m.moghadas2012@gmail.com>
Date: Thu, 8 Jul 2021 19:46:20 +0430
Subject: [PATCH] pathlib creating args.save

---
 train.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/train.py b/train.py
index 9e1cf63..159dde9 100644
--- a/train.py
+++ b/train.py
@@ -1,4 +1,5 @@
 import torch
+import pathlib
 import numpy as np
 import argparse
 import time
@@ -55,7 +56,8 @@ def main():
     if args.aptonly:
         supports = None
 
-
+    pathlib.Path(args.save).mkdir(parents=True, exist_ok=True)
+    print(f"Save directory: {args.save} has been created")
 
     engine = trainer(scaler, args.in_dim, args.seq_length, args.num_nodes, args.nhid, args.dropout,
                          args.learning_rate, args.weight_decay, device, supports, args.gcn_bool, args.addaptadj,