fix a typo

7 months ago · bac78657c1
parent 2e10eacddc
commit bac78657c1
2 changed files with 2 additions and 2 deletions
--- a/examples/YOLOv8-Action-Recognition/action_recognition.py
+++ b/examples/YOLOv8-Action-Recognition/action_recognition.py
@ -49,7 +49,7 @@ class TorchVisionVideoClassifier:
    if supports_mvitv2s:
        from torchvision.models.video import MViT_V2_S_Weights, mvit_v2_s

-    model_name_to_model_and_weights["mvit_v2_s"] = (mvit_v2_s, MViT_V2_S_Weights.DEFAULT)
+        model_name_to_model_and_weights["mvit_v2_s"] = (mvit_v2_s, MViT_V2_S_Weights.DEFAULT)

    def __init__(self, model_name: str, device: str or torch.device = ""):
        """
--- a/ultralytics/solutions/action_recognition.py
+++ b/ultralytics/solutions/action_recognition.py
@ -233,7 +233,7 @@ class TorchVisionVideoClassifier:
    if supports_mvitv2s:
        from torchvision.models.video import MViT_V2_S_Weights, mvit_v2_s

-    model_name_to_model_and_weights["mvit_v2_s"] = (mvit_v2_s, MViT_V2_S_Weights.DEFAULT)
+        model_name_to_model_and_weights["mvit_v2_s"] = (mvit_v2_s, MViT_V2_S_Weights.DEFAULT)

    def __init__(self, model_name: str, device: str or torch.device = ""):
        """