Thanks for the information!
That doesn’t really clear things up, so could you please print the shapes and values of each argument you are passing?
print(subject['image_name'])
print(image.cpu().detach().numpy().shape)
...