diff --git a/utils.py b/utils.py index 2f82405..7bbf74e 100644 --- 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)