From bf17659b7f0bd7951d544b383b448c8f39e6eaf9 Mon Sep 17 00:00:00 2001 From: Arjun Suresh Date: Tue, 30 Jan 2024 11:46:24 +0000 Subject: [PATCH] Added extra deps for nvidia resnet50 --- cm-mlops/script/reproduce-mlperf-inference-nvidia/_cm.yaml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/cm-mlops/script/reproduce-mlperf-inference-nvidia/_cm.yaml b/cm-mlops/script/reproduce-mlperf-inference-nvidia/_cm.yaml index caacaca668..6a3f615ae9 100644 --- a/cm-mlops/script/reproduce-mlperf-inference-nvidia/_cm.yaml +++ b/cm-mlops/script/reproduce-mlperf-inference-nvidia/_cm.yaml @@ -300,7 +300,7 @@ variations: env: CM_MODEL: resnet50 deps: - - tags: get,generic-python-lib,_graphsurgeon + - tags: get,generic-python-lib,_onnx-graphsurgeon - tags: get,generic-python-lib,_package.onnx version: 1.13.1 @@ -316,7 +316,7 @@ variations: - tags: get,generic-python-lib,_opencv-python - tags: get,generic-python-lib,_numpy - tags: get,generic-python-lib,_pycocotools - - tags: get,generic-python-lib,_graphsurgeon + - tags: get,generic-python-lib,_onnx-graphsurgeon - tags: get,generic-python-lib,_package.onnx version: 1.13.1