transunet_pytorch

transunet_pytorch copied to clipboard

how can i load vit‘s pretrained weight

1

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...

Soso-developer

I used the following config to successfully train a binary segmentation model on my dataset (640x480 pixel images) using your code: ``` cfg = EasyDict() cfg.batch_size = 2 cfg.epoch =...

bc-bytes