Create the inference log directory if not exists

ceshine · Oct 15, 2019 · 7ec86d9 · 7ec86d9
1 parent b6d09e5
commit 7ec86d9
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/yt8m/create_submission_from_memmaps.py b/yt8m/create_submission_from_memmaps.py
@@ -36,6 +36,8 @@ def main():
     arg('--model-names', nargs="+")
     args = parser.parse_args()
 
+    Path("data/cache/inference").mkdir(exist_ok=True, parents=True)
+
     model_dir = Path(args.model_dir)
     predictions = []
     with open(f"data/cache/inference/{datetime.now().strftime('log_%m%d_%H%M')}.txt", "w") as fout: