add model checkpointing to response selector

RasaHQ · Jul 15, 2020 · 32fe373 · 32fe373
1 parent f41a448
commit 32fe373
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/rasa/nlu/selectors/response_selector.py b/rasa/nlu/selectors/response_selector.py
@@ -79,6 +79,7 @@
     INTENT,
     DEFAULT_OPEN_UTTERANCE_TYPE,
     TEXT,
+    CHECKPOINT_MODEL,
 )
 
 from rasa.utils.tensorflow.model_data import RasaModelData
@@ -214,6 +215,8 @@ def required_components(cls) -> List[Type[Component]]:
         # Specify what features to use as sequence and sentence features
         # By default all features in the pipeline are used.
         FEATURIZERS: [],
+        # Perform model checkpointing
+        CHECKPOINT_MODEL: False,
     }
 
     def __init__(