logo
Browse Source

modify readme

Signed-off-by: gexy5 <xinyu.ge@zilliz.com>
main
gexy5 2 years ago
parent
commit
34d782d81d
  1. 2
      README.md
  2. 2
      __init__.py

2
README.md

@ -64,7 +64,7 @@ model_name='omnivore_swinT', skip_preprocess=False, classmap=None, topk=5)***
***model_name***: *str*
​ The name of pre-trained tsm model.
​ The name of pre-trained uniformer model.
​ Supported model names:
- omnivore_swinT

2
__init__.py

@ -16,4 +16,4 @@ from .omnivore import Omnivore
def omnivore(**kwargs):
return Omnivore(**kwargs)
return Omnivore(**kwargs)

Loading…
Cancel
Save