From af6ae5466e7c85ff2aa3a8cc6c2252300e0f3291 Mon Sep 17 00:00:00 2001 From: Alex Dautel Date: Tue, 23 Jul 2024 20:09:57 +0200 Subject: [PATCH] Fix deprecated np.float --- util/pos_embed.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/util/pos_embed.py b/util/pos_embed.py index 6acf8bd3b3..32d8f82275 100644 --- a/util/pos_embed.py +++ b/util/pos_embed.py @@ -8,9 +8,9 @@ # -------------------------------------------------------- import numpy as np - import torch + # -------------------------------------------------------- # 2D sine-cosine position embedding # References: @@ -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,)