Hi! Thanks for asking, and sorry you are having trouble.
Are you resizing the images to 1024x1024? One of the tradeoffs that this makes is that it is expecting only RGB images at that size. Thus, the network architecture is not very flexible.
Try using the torchvision.transforms.Resize with a size of (1024, 1024) on your input and see if that helps.
Also, I haven’t used this repo with the newer releases of PyTorch (I believe I developed it on 0.2 something) so you are bound to see some deprecation warnings.
Hope that helps!