Save input cfg file

.gitignore

@@ -92,3 +92,4 @@ website/pages/tutorials/*
 *-checkpoint.ipynb
 **/.ipynb_checkpoints
 **/.ipynb_checkpoints/**
+train_config.yaml

vissl/utils/hydra_config.py

@@ -7,6 +7,7 @@
 
 from omegaconf import DictConfig, OmegaConf
 from vissl.config import check_cfg_version
+from vissl.utils.io import save_file
 
 
 class AttrDict(dict):
@@ -86,6 +87,10 @@ def __dir__(self):
         return self.keys()
 
 
+def save_attrdict_to_disk(cfg, filename="train_config.yaml"):
+    save_file(cfg, filename)
+
+
 def convert_to_attrdict(cfg: DictConfig, cmdline_args: List[Any] = None):
     """
     Given the user input Hydra Config, and some command line input options
@@ -117,6 +122,7 @@ def convert_to_attrdict(cfg: DictConfig, cmdline_args: List[Any] = None):
     # assert the config and infer
     config = cfg.config
     assert_hydra_conf(config)
+    save_attrdict_to_disk(cfg)
     return cfg, config
 
 

vissl/utils/io.py

@@ -9,6 +9,7 @@
 from urllib.parse import urlparse
 
 import numpy as np
+import yaml
 from fvcore.common.download import download
 from fvcore.common.file_io import PathManager, file_lock
 from vissl.utils.slurm import get_slurm_dir
@@ -65,6 +66,11 @@ def save_file(data, filename):
         with PathManager.open(filename, "a") as fopen:
             fopen.write(json.dumps(data, sort_keys=True) + "\n")
             fopen.flush()
+    elif file_ext == ".yaml":
+        with PathManager.open(filename, "w") as fopen:
+            dump = yaml.dump(data)
+            fopen.write(dump)
+            fopen.flush()
     else:
         raise Exception(f"Saving {file_ext} is not supported yet")
     logging.info(f"Saved data to file: {filename}")