Peter Kováč
Results
1
comments of
Peter Kováč
I fixed this by adding `.detach()` before `.numpy()`. ```py displacement_map = opdict['displacement_map'][i].cpu().detach().numpy().squeeze() ```