Merge pull request #412 from mottodora/fix-molnet-example

make directory in molnet example if not exist
chainer · Dec 9, 2019 · 35f6709 · 35f6709
2 parents 3b6f55d + a7206d9
commit 35f6709
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/examples/molnet/train_molnet.py b/examples/molnet/train_molnet.py
@@ -222,6 +222,8 @@ def main():
     optimizer.setup(model)
 
     # Save model-related output to this directory.
+    if not os.path.exists(args.out):
+        os.makedirs(args.out)
     save_json(os.path.join(args.out, 'args.json'), vars(args))
     model_dir = os.path.join(args.out, os.path.basename(cache_dir))
     if not os.path.exists(model_dir):