Clean chain-inference

remfx/callbacks.py

@@ -64,9 +64,6 @@ class AudioCallback(Callback):
                     ]
                     for i, label in enumerate(effects_present_name):
                         self.log(f"{'_'.join(label)}", 0.0)
-                        # self.log(f"{effects}_{i}", label)
-                        # trainer.logger.experiment.log(
-                        # {f"effects_{i}": f"{'_'.join(label)}"}
                 else:
                     y = pl_module.model.sample(x)
             # Concat samples together for easier viewing in dashboard

scripts/chain_inference.py

@@ -15,7 +15,7 @@ def main(cfg: DictConfig):
         pl.seed_everything(cfg.seed)
     log.info(f"Instantiating datamodule <{cfg.datamodule._target_}>.")
     datamodule = hydra.utils.instantiate(cfg.datamodule, _convert_="partial")
-    log.info(f"Instantiating model <{cfg.model._target_}>.")
+    log.info("Instantiating Chain Inference Models")
     models = {}
     for effect in cfg.ckpts:
         model = hydra.utils.instantiate(cfg.ckpts[effect].model, _convert_="partial")