Register raises an error

[TheInfamousWayne]

I was trying to run the software version using the data_2 sample dataset provided. The style transfer and segmentation works. However, the Registration module throws the following error message:

I tried to typecast the line 66 in torch/utils/data/_utils/collate.py module to this return torch.as_tensor(batch, dtype=torch.uint8), but the error still exists. My guess is that the error is somewhere else.

I would be grateful if you could help with this.