tinyvision / SOLIDER-REID

MIT License
55 stars 12 forks source link

用预训练模型微调可以冻结主干网络吗 #11

Closed aliencaocao closed 1 year ago

aliencaocao commented 1 year ago

就只训练输出FC 用的market swin base

cwhgn commented 1 year ago

可以把希望冻住的参数的requires_grad设置成False。

aliencaocao commented 1 year ago

好的谢谢