add makedir

brycedrennan · Sep 23, 2021 · 96bc284 · 96bc284
1 parent 0e28386
commit 96bc284
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/download.py b/download.py
@@ -32,6 +32,8 @@ def save_response_content(response, destination):
 
 
 if __name__ == "__main__":
+    os.makedirs('./checkpoints', exist_ok=True)
+    os.makedirs('./examples', exist_ok=True)
 
     if not os.path.exists('./checkpoints/nyu.pt'):
         print('downloading the model trained on NYUv2...')