BUILD 929 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
licenses(["notice"])

package(
    default_visibility = [
        "//:__subpackages__",
    ],
)

py_library(
10
11
    name = "metric_functions",
    srcs = ["metric_functions.py"],
12
13
14
15
    srcs_version = "PY3",
)

py_library(
16
17
    name = "input_fn_reader",
    srcs = ["input_fn_reader.py"],
18
19
    srcs_version = "PY3",
    deps = [
20
        "//layers:projection_layers",
21
22
23
    ],
)

24
25
26
27
py_binary(
    name = "trainer",
    srcs = ["trainer.py"],
    python_version = "PY3",
28
    srcs_version = "PY3",
29
30
31
32
33
    deps = [
        ":input_fn_reader",
        ":metric_functions",
        "//models:prado",
    ],
34
35
36
)

py_binary(
37
38
    name = "export_to_tflite",
    srcs = ["export_to_tflite.py"],
39
40
41
42
43
    python_version = "PY3",
    srcs_version = "PY3",
    deps = [
        ":input_fn_reader",
        ":metric_functions",
44
45
46
47
        "//layers:base_layers",
        "//layers:projection_layers",
        "//models:prado",
        "//utils:tflite_utils",
48
49
    ],
)