Skip to content

Commit 1f3403d

Browse files
committed
WIP: Rewrite LSTM in PyTorch
Signed-off-by: format 2020.06.15 <github.com/ChrisCummins/format>
1 parent 724763c commit 1f3403d

File tree

10 files changed

+167
-179
lines changed

10 files changed

+167
-179
lines changed

WORKSPACE

-17
Original file line numberDiff line numberDiff line change
@@ -34,8 +34,6 @@ load(
3434

3535
pip_repositories()
3636

37-
# ProGraML pip requirements.
38-
3937
pip3_import(
4038
name = "programl_requirements",
4139
timeout = 3600,
@@ -49,21 +47,6 @@ load(
4947

5048
programl_pip_install()
5149

52-
# TensorFlow pip requirements.
53-
54-
pip3_import(
55-
name = "programl_tensorflow_requirements",
56-
timeout = 3600,
57-
requirements = "@programl//third_party/py/tensorflow:requirements.txt",
58-
)
59-
60-
load(
61-
"@programl_tensorflow_requirements//:requirements.bzl",
62-
programl_pip_install = "pip_install",
63-
)
64-
65-
programl_pip_install()
66-
6750
# Protobuf.
6851
pip3_import(
6952
name = "protobuf_py_deps",

programl/models/lstm/BUILD

+1-1
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ py_library(
2727
"//programl/proto:epoch_py",
2828
"//third_party/py/labm8",
2929
"//third_party/py/numpy",
30-
"//third_party/py/tensorflow",
30+
"//third_party/py/torch",
3131
],
3232
)
3333

0 commit comments

Comments
 (0)