Merge pull request hkchengrex#69 from linhandev/patch-1

Change np.float to np.float32
kristianmk · Mar 8, 2023 · 0909938 · 0909938
2 parents af90122 + 346c078
commit 0909938
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/inference/interact/fbrs/inference/predictors/brs_functors.py b/inference/interact/fbrs/inference/predictors/brs_functors.py
@@ -72,7 +72,7 @@ def __call__(self, x):
         self._last_mask = current_mask
 
         loss.backward()
-        f_grad = opt_params.grad.cpu().numpy().ravel().astype(np.float)
+        f_grad = opt_params.grad.cpu().numpy().ravel().astype(np.float32)
 
         return [f_val, f_grad]