Update difpoint/inference.py

difpoint/inference.py

@@ -376,7 +376,7 @@ class Inferencer(object):
 
             item_dct = {
                 'scale': x_d_i_info['scale'][i].cpu().numpy().astype(np.float32),
-                'R_d': R_d_i.astype(np.float32),
+                'R_d': R_d_i.cpu().numpy().astype(np.float32),
                 'exp': x_d_i_info['exp'][i].reshape(1, 21, -1).cpu().numpy().astype(np.float32),
                 't': x_d_i_info['t'][i].cpu().numpy().astype(np.float32),
             }