ultralytics 8.1.9 replace .size(0) with .shape[0] (#7957)

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-02-01 23:33:02 +01:00 · 2024-02-01 23:33:02 +01:00 · 70d4a3752e
commit 70d4a3752e
parent 66b32bb4dd
15 changed files with 42 additions and 22 deletions
--- a/ultralytics/models/sam/modules/tiny_encoder.py
+++ b/ultralytics/models/sam/modules/tiny_encoder.py
@ -732,7 +732,7 @@ class TinyViT(nn.Module):
        for i in range(start_i, len(self.layers)):
            layer = self.layers[i]
            x = layer(x)
-        B, _, C = x.size()
+        B, _, C = x.shape
        x = x.view(B, 64, 64, C)
        x = x.permute(0, 3, 1, 2)
        return self.neck(x)