Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

Appearance settings

Commit 169b081

Browse filesBrowse files
committed
[ConvNets/TF1] Added Triton for ResNet
1 parent 7bdfc81 commit 169b081
Copy full SHA for 169b081

File tree

Expand file treeCollapse file tree

118 files changed

+100310
-996
lines changed
Open diff view settings
Filter options

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.
Dismiss banner
Expand file treeCollapse file tree

118 files changed

+100310
-996
lines changed
Open diff view settings
Collapse file

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

Copy file name to clipboardExpand all lines: TensorFlow/Classification/ConvNets/.style.yapf
+2-2Lines 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
Collapse file
+25-3Lines 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 . .
Collapse file

‎TensorFlow/Classification/ConvNets/README.md‎

Copy file name to clipboardExpand all lines: TensorFlow/Classification/ConvNets/README.md
+2-2Lines changed: 2 additions & 2 deletions
  • Display the source diff
  • Display the rich diff
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 (0)
Morty Proxy This is a proxified and sanitized view of the page, visit original site.