Clean up

LarsKue · Oct 27, 2023 · a17f51e · a17f51e
1 parent 927ee5f
commit a17f51e
Showing 1 changed file with 0 additions and 5 deletions.
diff --git a/src/lightning_trainable/trainable/trainable.py b/src/lightning_trainable/trainable/trainable.py
@@ -1,6 +1,5 @@
 import lightning
 import os
-import pathlib
 import torch
 
 from copy import deepcopy
@@ -62,10 +61,6 @@ def compute_metrics(self, batch, batch_idx) -> dict:
         """
         raise NotImplementedError
 
-    def on_train_start(self):
-        # TODO: get metrics to track from compute_metrics
-        self.logger.log_hyperparams(self.hparams, ...)
-
     def training_step(self, batch, batch_idx):
         try:
             metrics = self.compute_metrics(batch, batch_idx)