From a49c9d7bdf6426f09428b499d0b1750b15bb66a5 Mon Sep 17 00:00:00 2001 From: yh_cc Date: Tue, 10 May 2022 22:10:56 +0800 Subject: [PATCH] =?UTF-8?q?fitlog=5Fcallback=E4=BF=AE=E5=A4=8D?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- fastNLP/core/callbacks/fitlog_callback.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/fastNLP/core/callbacks/fitlog_callback.py b/fastNLP/core/callbacks/fitlog_callback.py index 39fbb684..a8b87c1a 100644 --- a/fastNLP/core/callbacks/fitlog_callback.py +++ b/fastNLP/core/callbacks/fitlog_callback.py @@ -1,8 +1,10 @@ __all__ = [ 'FitlogCallback' ] -from fastNLP import HasMonitorCallback -import fitlog +from .has_monitor_callback import HasMonitorCallback +from ...envs import _module_available +if _module_available('fitlog'): + import fitlog class FitlogCallback(HasMonitorCallback): @@ -25,6 +27,8 @@ class FitlogCallback(HasMonitorCallback): :param log_exception: 是否记录 ``exception`` 。 :param log_loss_every: 多少个 ``batch`` 记录一次 loss 到 ``fitlog`` 中。 """ + assert _module_available('fitlog'), "fitlog is not installed." + super().__init__(monitor=monitor, larger_better=larger_better) self.log_exception = log_exception self.log_loss_every = log_loss_every