torchgeo icon indicating copy to clipboard operation
torchgeo copied to clipboard

Adapt segmentation trainer to work with ViT

Open nilsleh opened this issue 1 year ago • 3 comments

nilsleh avatar Jun 06 '23 13:06 nilsleh

I assume this will also be needed in PixelwiseRegressionTask?

adamjstewart avatar Jun 09 '23 17:06 adamjstewart

This is how I have hacked it so far.

nilsleh avatar Jun 09 '23 18:06 nilsleh

Don't know why tests are failing but we need to fix that

adamjstewart avatar Jun 09 '23 21:06 adamjstewart