Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/megatron-lm.git" did not exist on "f5cf2e4270c6159061b7db505673f4d14b67d033"
Commit d0d99252 authored by Lukasz Kaiser's avatar Lukasz Kaiser
Browse files

Improve open-source Neural GPU code.

parent 90d6e3b9
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 41 deletions
neural_gpu/BUILD deleted 100644 → 0
View file @ 90d6e3b9
py_library(
name = "data_utils",
srcs = [
"data_utils.py",
],
deps = [
"//file/colossus/public:cns",
"//third_party/py/numpy",
"//third_party/py/tensorflow",
],
)
py_library(
name = "neural_gpu",
srcs = [
"neural_gpu.py",
],
deps = [
":data_utils",
"//third_party/py/numpy",
"//third_party/py/tensorflow",
],
)
py_binary(
name = "neural_gpu_trainer",
srcs = [
"neural_gpu_trainer.py",
],
launcher = "//devtools/python/launcher",
malloc = "//tcmalloc:tcmalloc_or_debug",
deps = [
":neural_gpu",
"//file/colossus/public:cns",
"//net/proto2/python/public:use_fast_cpp_protos",
"//third_party/py/Tkinter",
"//third_party/py/matplotlib",
"//third_party/py/numpy",
"//third_party/py/tensorflow",
],
)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment