From bed0739f580b4553b56ca3332ca5af011b6f6b4e Mon Sep 17 00:00:00 2001 From: Srikumar Sastry Date: Wed, 20 Dec 2023 15:15:02 -0600 Subject: [PATCH] Update pos_embed.py Fixed `np.float` to `np.float32` --- MAEPretrain_SceneClassification/util/pos_embed.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/MAEPretrain_SceneClassification/util/pos_embed.py b/MAEPretrain_SceneClassification/util/pos_embed.py index 6acf8bd..bcafafe 100644 --- a/MAEPretrain_SceneClassification/util/pos_embed.py +++ b/MAEPretrain_SceneClassification/util/pos_embed.py @@ -53,7 +53,7 @@ def get_1d_sincos_pos_embed_from_grid(embed_dim, pos): out: (M, D) """ assert embed_dim % 2 == 0 - omega = np.arange(embed_dim // 2, dtype=np.float) + omega = np.arange(embed_dim // 2, dtype=np.float32) omega /= embed_dim / 2. omega = 1. / 10000**omega # (D/2,)