Include model_dir in estimator

MetcalfeTom · MetcalfeTom · commit 9359f274aa33 · 2019-02-28T16:57:26.000+01:00
diff --git a/rasa_nlu/featurizers/bert_featurizer.py b/rasa_nlu/featurizers/bert_featurizer.py
@@ -71,6 +71,7 @@ def __init__(self, component_config=None):
            use_tpu=False,
            model_fn=model_fn,
            config=run_config,
+           model_dir='/tmp/bert_model',
            predict_batch_size=8)
 
     def train(self, training_data, config, **kwargs):
