RNNGJit (#597)

Summary: Pull Request resolved: #597 RNNG model using JIT, e2e works in training time now Reviewed By: wowitsmrinal Differential Revision: D15001036 fbshipit-source-id: 173246fc0386282e804b6df9aae333135d57f6f2
facebookresearch · Jun 5, 2019 · 358ecec · 358ecec
1 parent 0f79ca3
commit 358ecec
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"]