Browse Source
Fix typo
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
main
1 changed files with
3 additions and
3 deletions
-
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) |
|
|