diff --git a/fastNLP/__init__.py b/fastNLP/__init__.py index ce68d5f3..85b9831e 100644 --- a/fastNLP/__init__.py +++ b/fastNLP/__init__.py @@ -69,6 +69,8 @@ __all__ = [ "BCELoss", "NLLLoss", "LossInForward", + "LossBase", + "CMRC2018Loss", "cache_results",