Skip to content

Commit 169b081

Browse files
nv-kkudrynskinv-kkudrynski
committed
[ConvNets/TF1] Added Triton for ResNet
1 parent 7bdfc81 commit 169b081

118 files changed

Lines changed: 100310 additions & 996 deletions

File tree

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

‎TensorFlow/Classification/ConvNets/.style.yapf‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ allow_multiline_lambdas = True
3232
# # <------ this blank line
3333
# def method():
3434
# pass
35-
blank_line_before_nested_class_or_def = True
35+
blank_line_before_nested_class_or_def = False
3636

3737
# Insert a blank line before a module docstring.
3838
blank_line_before_module_docstring = True
@@ -83,7 +83,7 @@ continuation_indent_width = 4
8383
# start_ts=now()-timedelta(days=3),
8484
# end_ts=now(),
8585
# ) # <--- this bracket is dedented and on a separate line
86-
dedent_closing_brackets = True
86+
dedent_closing_brackets = False
8787

8888
# Disable the heuristic which places each list element on a separate line if the list is comma-terminated.
8989
disable_ending_comma_heuristic = false

‎TensorFlow/Classification/ConvNets/Dockerfile‎

Lines changed: 25 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,30 @@
1-
ARG FROM_IMAGE_NAME=nvcr.io/nvidia/tensorflow:20.06-tf1-py3
1+
ARG FROM_IMAGE_NAME=nvcr.io/nvidia/tensorflow:20.12-tf1-py3
2+
ARG TRITON_CLIENT_IMAGE_NAME=nvcr.io/nvidia/tritonserver:20.12-py3-sdk
3+
FROM ${TRITON_CLIENT_IMAGE_NAME} as triton-client
24
FROM ${FROM_IMAGE_NAME}
35

6+
# Install perf_client required library
7+
RUN apt-get update && \
8+
apt-get install -y libb64-dev libb64-0d && \
9+
apt-get clean && \
10+
rm -rf /var/lib/apt/lists/*
11+
12+
# Install Triton Client PythonAPI and copy Perf Client
13+
COPY --from=triton-client /workspace/install/ /workspace/install/
14+
ENV LD_LIBRARY_PATH /workspace/install/lib:${LD_LIBRARY_PATH}
15+
RUN find /workspace/install/python/ -iname triton*manylinux*.whl -exec pip install {}[all] \;
16+
17+
# Setup environmnent variables to access Triton Client lib and bin
18+
ENV PATH /workspace/install/bin:${PATH}
19+
20+
ENV PYTHONPATH /workspace/rn50v15_tf
21+
WORKDIR /workspace/rn50v15_tf
22+
23+
RUN pip uninstall -y typing
24+
425
ADD requirements.txt .
26+
ADD triton/requirements.txt triton/requirements.txt
527
RUN pip install -r requirements.txt
28+
RUN pip install -r triton/requirements.txt
629

7-
ADD . /workspace/rn50v15_tf
8-
WORKDIR /workspace/rn50v15_tf
30+
ADD . .

‎TensorFlow/Classification/ConvNets/README.md‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ were averaged over an entire training epoch.
5151
The specific training script that was run is documented
5252
in the corresponding model's README.
5353

54-
The following table shows the training accuracy results of the
54+
The following table shows the training performance results of the
5555
three classification models side-by-side.
5656

5757

@@ -71,7 +71,7 @@ were averaged over an entire training epoch.
7171
The specific training script that was run is documented
7272
in the corresponding model's README.
7373

74-
The following table shows the training accuracy results of the
74+
The following table shows the training performance results of the
7575
three classification models side-by-side.
7676

7777

0 commit comments

Comments
 (0)