Create wandb directory if it doesn't exist

Author: asogaard

examples/train_model.py

@@ -1,4 +1,4 @@
-import os.path
+import os
 
 from pytorch_lightning import Trainer
 from pytorch_lightning.callbacks import EarlyStopping
@@ -41,6 +41,9 @@
     log_model=True,
 )
 
+# -- Ensure that custom output directory exists
+os.makedirs(wandb_logger.save_dir, exists_ok=True)
+
 
 # Main function definition
 def main():