transunet_pytorch icon indicating copy to clipboard operation
transunet_pytorch copied to clipboard

how can i load vit‘s pretrained weight

Open Soso-developer opened this issue 1 year ago • 1 comments

I refer to your code to put vit block into my u-net,it works,thank you a lot
But another problem is about loading pretrained weight. In encoder I use vgg or resnet and load pretrained weight, but I dont know how to load vit weight on its blocks.
Hope can get your help!

Soso-developer avatar May 05 '23 09:05 Soso-developer

I met the same problem

Data-Designer avatar Sep 16 '23 01:09 Data-Designer