From b6507daf891cc4e9fcaf3d5c8ddc816d3bfcf5ec Mon Sep 17 00:00:00 2001 From: davinnovation Date: Sun, 5 Jul 2020 12:55:22 +0900 Subject: [PATCH] add metrics to pl module (#2506) --- pytorch_lightning/__init__.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/pytorch_lightning/__init__.py b/pytorch_lightning/__init__.py index 9a6469fc72..a03f2b9e80 100644 --- a/pytorch_lightning/__init__.py +++ b/pytorch_lightning/__init__.py @@ -54,6 +54,7 @@ else: from pytorch_lightning.callbacks import Callback from pytorch_lightning.trainer import Trainer from pytorch_lightning.utilities.seed import seed_everything + from pytorch_lightning import metrics __all__ = [ 'Trainer', @@ -61,6 +62,7 @@ else: 'Callback', 'data_loader', 'seed_everything', + 'metrics' ] # necessary for regular bolts imports. Skip exception since bolts is not always installed