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 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.