Thanks, I tried that, still the same issue. In fact, I tried all different combinations
device=torch.device('cuda:1'), device='cuda:1' (when using output_device=device, and transferring models and features)
as well
as output_device=1 Still the same issue.