Update utils.py

KaiservonAfrika · Oct 11, 2022 · a2c202e · a2c202e
1 parent c38b5e6
commit a2c202e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/utils.py b/utils.py
@@ -149,8 +149,8 @@ def get_hparams(init=True):
                       help='Model name')
 
   args = parser.parse_args()
-  #model_dir = os.path.join("../drive/MyDrive", args.model)
-  model_dir = os.path.join("./logs", args.model)
+  model_dir = os.path.join("../drive/MyDrive", args.model)
+  #model_dir = os.path.join("./logs", args.model)
 
   if not os.path.exists(model_dir):
     os.makedirs(model_dir)