Browse Source
Fix typo
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
main
1 changed files with
3 additions and
3 deletions
-
pytorchvideo.py
|
|
@ -38,8 +38,8 @@ class PytorchVideo(NNOperator): |
|
|
- mvit_base_32x3 |
|
|
- mvit_base_32x3 |
|
|
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). |
|
|
""" |
|
|
""" |
|
|
@ -49,7 +49,7 @@ class PytorchVideo(NNOperator): |
|
|
model_name: str = 'x3d_xs', |
|
|
model_name: str = 'x3d_xs', |
|
|
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, |
|
|
) -> None: |
|
|
) -> None: |
|
|
super().__init__(framework=framework) |
|
|
super().__init__(framework=framework) |
|
|
|
