From a0fca500be60225d54e30ac424d8e3b585176353 Mon Sep 17 00:00:00 2001 From: "zhanning.gzn" Date: Sat, 6 Aug 2022 00:30:24 +0800 Subject: [PATCH] Merge branch cv/image-classification-emian into master Title: fix bugs - Inputs to input for ClassificationModel Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/9660908 --- modelscope/models/cv/image_classification/mmcls_model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/modelscope/models/cv/image_classification/mmcls_model.py b/modelscope/models/cv/image_classification/mmcls_model.py index 52c0568b..6a65656e 100644 --- a/modelscope/models/cv/image_classification/mmcls_model.py +++ b/modelscope/models/cv/image_classification/mmcls_model.py @@ -27,9 +27,9 @@ class ClassificationModel(TorchModel): self.load_pretrained_checkpoint() - def forward(self, Inputs): + def forward(self, inputs): - return self.cls_model(**Inputs) + return self.cls_model(**inputs) def load_pretrained_checkpoint(self): import mmcv