From 12f0d364a91b03e0c60e9d49c2107e85ff278e2a Mon Sep 17 00:00:00 2001 From: sheele41 Date: Wed, 3 Nov 2021 18:02:28 +0900 Subject: [PATCH] modify runtime/image_classification/models/resnet50/gpus=2/__init__.py --- runtime/image_classification/models/resnet50/gpus=2/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/runtime/image_classification/models/resnet50/gpus=2/__init__.py b/runtime/image_classification/models/resnet50/gpus=2/__init__.py index 49e5a26..4d1edcd 100644 --- a/runtime/image_classification/models/resnet50/gpus=2/__init__.py +++ b/runtime/image_classification/models/resnet50/gpus=2/__init__.py @@ -12,7 +12,7 @@ def arch(): def model(criterion): return [ - (Stage0(), ["input"], ["out0", "out1"]), + (Stage0(), ["input0"], ["out0", "out1"]), (Stage1(), ["out0", "out1"], ["out3", "out2"]), (Stage2(), ["out3", "out2"], ["out4", "out5"]), (Stage3(), ["out4", "out5"], ["out6"]),