add fitlogcallback to init

fastNLP/core/__init__.py

@@ -14,6 +14,7 @@ __all__ = [
     "TorchGradClipCallback",
     "ResultsMonitor",
     'HasMonitorCallback',
+    "FitlogCallback",
 
     # collators
     'Collator',

fastNLP/core/callbacks/__init__.py

@@ -17,7 +17,9 @@ __all__ = [
     "TorchGradClipCallback",
 
     "ResultsMonitor",
-    'HasMonitorCallback'
+    'HasMonitorCallback',
+
+    "FitlogCallback"
 ]
 
 
@@ -32,4 +34,5 @@ from .early_stop_callback import EarlyStopCallback
 from .torch_callbacks import *
 from .more_evaluate_callback import MoreEvaluateCallback
 from .has_monitor_callback import ResultsMonitor, HasMonitorCallback
+from .fitlog_callback import FitlogCallback
 

fastNLP/core/callbacks/fitlog_callback.py

@@ -1,3 +1,6 @@
+__all__ = [
+    'FitlogCallback'
+]
 from fastNLP import HasMonitorCallback
 import fitlog