Restore wandblogger

cfg/config.yaml

@@ -2,6 +2,7 @@ defaults:
   - _self_
   - model: null
   - effects: all
+  - logger: csv
 
 seed: 12345
 train: True
@@ -105,19 +106,6 @@ datamodule:
   pin_memory: True
   persistent_workers: True
 
-# logger:
-#   _target_: pytorch_lightning.loggers.WandbLogger
-#   project: ${oc.env:WANDB_PROJECT}
-#   entity: ${oc.env:WANDB_ENTITY}
-#   # offline: False  # set True to store all logs only locally
-#   job_type: "train"
-#   group: ""
-#   save_dir: "."
-#   log_model: True
-logger:
-  _target_: pytorch_lightning.loggers.CSVLogger
-  save_dir: "."
-  version: ${now:%Y-%m-%d-%H-%M-%S}
 
 trainer:
   _target_: pytorch_lightning.Trainer

cfg/logger/csv.yaml

@@ -0,0 +1,5 @@
+# @package _global_
+logger:
+  _target_: pytorch_lightning.loggers.CSVLogger
+  save_dir: "."
+  version: ${now:%Y-%m-%d-%H-%M-%S}

cfg/logger/wandb.yaml

@@ -0,0 +1,10 @@
+# @package _global_
+logger:
+  _target_: pytorch_lightning.loggers.WandbLogger
+  project: ${oc.env:WANDB_PROJECT}
+  entity: ${oc.env:WANDB_ENTITY}
+  # offline: False  # set True to store all logs only locally
+  job_type: "train"
+  group: ""
+  save_dir: "."
+  log_model: True

remfx/callbacks.py

@@ -50,7 +50,7 @@ class AudioCallback(Callback):
                 # Avoids circular import
                 from remfx.models import RemFXChainInference
 
-                if type(pl_module) == RemFXChainInference:
+                if isinstance(pl_module, RemFXChainInference):
                     y = pl_module.sample(batch)
                     effects_present_name = [
                         [