search

bubbliiiing / segformer-pytorch

这是一个segformer-pytorch的源码，可以用于训练自己的模型。
MIT License
282 stars 36 forks source link

只加载【backbone】权重，设置pretrain=true model_path=''问题 #29

Closed rtexpio closed 9 months ago

rtexpio commented 9 months ago

只想加载backbone权重，设置pretrain=true model_path=''后发现只调用download_weights()方法，而其中load_state_dict_from_url()只是下载权重，因为model_path='' 所以 if model_path != '':后面的判断权值过程没执行。。。。。因此train之前没发现有加载权重的过程，疑惑中