Handling cfloat issue identified at mbzuai-metaverse#20

jeromeyers · Feb 7, 2024 · 5103222 · 5103222
1 parent f7be590
commit 5103222
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.cfloat)
 
         return [f_val, f_grad]