transunet_pytorch
transunet_pytorch copied to clipboard
how can i load vit‘s pretrained weight
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!
I met the same problem