Browse Source

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
master
zhanning.gzn wenmeng.zwm 3 years ago
parent
commit
a0fca500be
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      modelscope/models/cv/image_classification/mmcls_model.py

+ 2
- 2
modelscope/models/cv/image_classification/mmcls_model.py View File

@@ -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


Loading…
Cancel
Save