From 7e8e54e84c63171e748bbf09516fd517e6821ace Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Sat, 20 Mar 2021 18:09:06 +0100
Subject: Inital commit for refactoring to lightning

---
 src/training/trainer/util.py | 28 ----------------------------
 1 file changed, 28 deletions(-)
 delete mode 100644 src/training/trainer/util.py

(limited to 'src/training/trainer/util.py')

diff --git a/src/training/trainer/util.py b/src/training/trainer/util.py
deleted file mode 100644
index 7cf1b45..0000000
--- a/src/training/trainer/util.py
+++ /dev/null
@@ -1,28 +0,0 @@
-"""Utility functions for training neural networks."""
-from typing import Dict, Optional
-
-from loguru import logger
-
-
-def log_val_metric(metrics_mean: Dict, epoch: Optional[int] = None) -> None:
-    """Logging of val metrics to file/terminal."""
-    log_str = "Validation metrics " + (f"at epoch {epoch} - " if epoch else " - ")
-    logger.debug(log_str + " - ".join(f"{k}: {v:.4f}" for k, v in metrics_mean.items()))
-
-
-class RunningAverage:
-    """Maintains a running average."""
-
-    def __init__(self) -> None:
-        """Initializes the parameters."""
-        self.steps = 0
-        self.total = 0
-
-    def update(self, val: float) -> None:
-        """Updates the parameters."""
-        self.total += val
-        self.steps += 1
-
-    def __call__(self) -> float:
-        """Computes the running average."""
-        return self.total / float(self.steps)
-- 
cgit v1.2.3-70-g09d2