From 287da379aaa8772c80a9a7a50c0f3af6820d2640 Mon Sep 17 00:00:00 2001 From: The-truthh <821372701@qq.com> Date: Tue, 12 Mar 2024 20:23:17 +0800 Subject: [PATCH] fix: fix the bug of ResizeBilinear for pvt --- mindcv/models/layers/compatibility.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindcv/models/layers/compatibility.py b/mindcv/models/layers/compatibility.py index 9df51e173..8aecbae77 100644 --- a/mindcv/models/layers/compatibility.py +++ b/mindcv/models/layers/compatibility.py @@ -89,7 +89,7 @@ def construct(self, x): class ResizeBilinear(nn.Cell): def __init__(self, size, align_corners=False, half_pixel_centers=False): super().__init__() - if hasattr(nn, "ResizeBilinearV2"): + if hasattr(ops, "ResizeBilinearV2"): self.resize_bilinear = ops.ResizeBilinearV2( align_corners=align_corners, half_pixel_centers=half_pixel_centers )