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,)