forward arguments for naming #117

point-cloud-radar · Jul 3, 2023 · d5b1cf8 · d5b1cf8
1 parent f95716a
commit d5b1cf8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/bird_cloud_gnn/callback.py b/bird_cloud_gnn/callback.py
@@ -5,8 +5,8 @@
 class TensorboardCallback:
     """Callback to populate Tensorboard"""
 
-    def __init__(self):
-        self.writer = SummaryWriter()
+    def __init__(self, **kwargs):
+        self.writer = SummaryWriter(**kwargs)
 
     def __call__(self, epoch_values):
         epoch = epoch_values["epoch"]