fit jit test

alibaba · Jul 18, 2024 · a392bb0 · a392bb0
1 parent bf3cf8b
commit a392bb0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/easycv/models/classification/classification.py b/easycv/models/classification/classification.py
@@ -151,6 +151,7 @@ def forward_backbone(self, img: torch.Tensor) -> List[torch.Tensor]:
         x = self.backbone(img)
         return x
 
+    @torch.jit.unused
     def forward_onnx(self, img: torch.Tensor) -> Dict[str, torch.Tensor]:
         """
             forward_onnx means generate prob from image only support one neck  + one head