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: fddac1d85dd9c1725d3b6fe8d20ca0146d0c9755
seayoung1112 · May 24, 2019 · 543925c · 543925c
1 parent 4095d45
commit 543925c
Show file tree

Hide file tree

Showing 10 changed files with 771 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"]