WIP: Rewrite LSTM in PyTorch

Signed-off-by: format 2020.06.15 <github.com/ChrisCummins/format>
ChrisCummins · Aug 18, 2020 · 1f3403d · 1f3403d
1 parent 724763c
commit 1f3403d
Show file tree

Hide file tree

Showing 10 changed files with 167 additions and 179 deletions.
diff --git a/WORKSPACE b/WORKSPACE
@@ -34,8 +34,6 @@ load(
 
 pip_repositories()
 
-# ProGraML pip requirements.
-
 pip3_import(
     name = "programl_requirements",
     timeout = 3600,
@@ -49,21 +47,6 @@ load(
 
 programl_pip_install()
 
-# TensorFlow pip requirements.
-
-pip3_import(
-    name = "programl_tensorflow_requirements",
-    timeout = 3600,
-    requirements = "@programl//third_party/py/tensorflow:requirements.txt",
-)
-
-load(
-    "@programl_tensorflow_requirements//:requirements.bzl",
-    programl_pip_install = "pip_install",
-)
-
-programl_pip_install()
-
 # Protobuf.
 pip3_import(
     name = "protobuf_py_deps",

diff --git a/programl/models/lstm/BUILD b/programl/models/lstm/BUILD
@@ -27,7 +27,7 @@ py_library(
         "//programl/proto:epoch_py",
         "//third_party/py/labm8",
         "//third_party/py/numpy",
-        "//third_party/py/tensorflow",
+        "//third_party/py/torch",
     ],
 )