Refactor TimmViTEncoder to use new forward_intermediates method from timm

isaaccorley / torchseg

Segmentation models with pretrained backbones. PyTorch.

MIT License

98 stars 8 forks source link

Open isaaccorley opened 5 months ago

isaaccorley commented 5 months ago

Now that timm has support for intermediate feature extraction we should refactor to use this method instead of our custom method. See https://github.com/huggingface/pytorch-image-models/pull/2136