torchgeo
torchgeo copied to clipboard
Adapt segmentation trainer to work with ViT
I assume this will also be needed in PixelwiseRegressionTask?
This is how I have hacked it so far.
Don't know why tests are failing but we need to fix that