add pl_logger to default hyperparams dict

mittagessen · Feb 21, 2023 · d81e898 · d81e898
1 parent 6baa9b3
commit d81e898
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/kraken/lib/default_specs.py b/kraken/lib/default_specs.py
@@ -49,6 +49,8 @@
                                      'num_negatives': 100,
                                      'logit_temp': 0.1,
                                      'warmup': 32000,
+                                     # loggers
+                                     'pl_logger': None,
                                      }
 
 RECOGNITION_HYPER_PARAMS = {'pad': 16,
@@ -79,6 +81,8 @@
                             'cos_t_max': 50,
                             'warmup': 0,
                             'freeze_backbone': 0,
+                            # loggers
+                            'pl_logger': None,
                             }
 
 SEGMENTATION_HYPER_PARAMS = {'line_width': 8,
@@ -106,4 +110,6 @@
                              # cosine
                              'cos_t_max': 50,
                              'warmup': 0,
+                             # loggers
+                             'pl_logger': None,
                              }