search

lucidrains / segformer-pytorch

Implementation of Segformer, Attention + MLP neural network for segmentation, in Pytorch
MIT License
342 stars 43 forks source link

how to use pre_trained weights? #12

Open dmndxld opened 1 year ago

dmndxld commented 1 year ago

Hi, your code implementation helped me a lot! I am doing a new segmentation task now, and I want to use pre-trained network weights like imagenet, how can I modify the code? Thanks!