Looking at the traceback, I believe the error is associated with torch.squeeze(Delta[:,i:i+1,:])). However, you haven’t shared the code. It could also be due to your version of PyTorch. As some users have reported the same error previously. Have a look at https://github.com/chrisdxie/uois/issues/2 3
I would suggest you to share the piece of code associated with slicing the Delta tensor.