Merge pull request #130 from geometric-intelligence/logger

Allow all hp logging

.gitignore

@@ -174,6 +174,7 @@ cython_debug/
 
 # Data
 /datasets/
+/data/
 /notebooks/*.csv
 notebooks/tmp
 *.pickle

topobenchmark/utils/logging_utils.py

@@ -35,8 +35,6 @@ def log_hyperparameters(object_dict: dict[str, Any]) -> None:
         log.warning("Logger not found! Skipping hyperparameter logging...")
         return
 
-    hparams["model"] = cfg["model"]
-
     # save number of model parameters
     hparams["model/params/total"] = sum(p.numel() for p in model.parameters())
     hparams["model/params/trainable"] = sum(
@@ -46,17 +44,8 @@ def log_hyperparameters(object_dict: dict[str, Any]) -> None:
         p.numel() for p in model.parameters() if not p.requires_grad
     )
 
-    hparams["dataset"] = cfg["dataset"]
-    hparams["trainer"] = cfg["trainer"]
-
-    hparams["callbacks"] = cfg.get("callbacks")
-    hparams["extras"] = cfg.get("extras")
-
-    hparams["task_name"] = cfg.get("task_name")
-    hparams["tags"] = cfg.get("tags")
-    hparams["ckpt_path"] = cfg.get("ckpt_path")
-    hparams["seed"] = cfg.get("seed")
-    hparams["paths"] = cfg.get("paths")
+    for key in cfg:
+        hparams[key] = cfg[key]
 
     # send hparams to all loggers
     for logger in trainer.loggers: