save configuration to wandb (#217)

* save configuration to wandb

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* save to top level wandb folder

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

pvnet/training.py

@@ -121,6 +121,14 @@ def train(config: DictConfig) -> Optional[float]:
 
                 assert os.path.isfile(data_config), f"Data config file not found: {data_config}"
                 shutil.copyfile(data_config, f"{callback.dirpath}/data_config.yaml")
+
+                # upload configuration up to wandb
+                OmegaConf.save(config, "./experiment_config.yaml")
+                wandb_logger.experiment.save(
+                    f"{callback.dirpath}/data_config.yaml", callback.dirpath
+                )
+                wandb_logger.experiment.save("./experiment_config.yaml")
+
                 break
 
     should_pretrain = False