logo
Browse Source

Fix typo

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
main
Jael Gu 2 years ago
parent
commit
7c52483f25
  1. 6
      timesformer.py

6
timesformer.py

@ -28,8 +28,8 @@ class Timesformer(NNOperator):
- timesformer_k400_8x224 - timesformer_k400_8x224
skip_preprocess (`str`): skip_preprocess (`str`):
Flag to skip video transforms. Flag to skip video transforms.
classmap (`str=None`):
Path of the json file to match class names.
classmap (`dict=None`):
The dictionary maps classes to integers.
topk (`int=5`): topk (`int=5`):
The number of classification labels to be returned (ordered by possibility from high to low). The number of classification labels to be returned (ordered by possibility from high to low).
""" """
@ -37,7 +37,7 @@ class Timesformer(NNOperator):
model_name: str = 'timesformer_k400_8x224', model_name: str = 'timesformer_k400_8x224',
framework: str = 'pytorch', framework: str = 'pytorch',
skip_preprocess: bool = False, skip_preprocess: bool = False,
classmap: str = None,
classmap: dict = None,
topk: int = 5, topk: int = 5,
): ):
super().__init__(framework=framework) super().__init__(framework=framework)

Loading…
Cancel
Save