RNNGJit (facebookresearch#597)

Summary: Pull Request resolved: facebookresearch#597 RNNG model using JIT, e2e works in training time now Reviewed By: wowitsmrinal Differential Revision: D15001036 fbshipit-source-id: f1f278fe3b84733a965a738fe2f148a9c2c8133e
seayoung1112 · May 30, 2019 · 665fcfe · 665fcfe
1 parent cd8f057
commit 665fcfe
Show file tree

Hide file tree

Showing 10 changed files with 765 additions and 27 deletions.
diff --git a/pytext/data/compositional_data_handler.py b/pytext/data/compositional_data_handler.py
@@ -156,7 +156,7 @@ def _test_input_from_batch(self, batch):
             text_input[0],
             text_input[1],
             getattr(batch, DatasetFieldName.DICT_FIELD, None),
-            None,
+            [],
             getattr(batch, DatasetFieldName.CONTEXTUAL_TOKEN_EMBEDDING, None),
             1,
             1,

diff --git a/pytext/models/semantic_parsers/rnng/jit/__init__.py b/pytext/models/semantic_parsers/rnng/jit/__init__.py
@@ -0,0 +1,6 @@
+#!/usr/bin/env python3
+# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
+from .model import RNNGModel, RNNGParserJIT
+
+
+__all__ = ["RNNGParserJIT", "RNNGModel"]