pretrained_dict = torch.load("./senet/seresnet50-60a8950a85b2b.pkl") model_dict = net.state_dict() # 重新制作預(yù)訓(xùn)練的權(quán)重,主要是減去參數(shù)不匹配的層,樓主這邊層名為“fc” pretrained_dict = {k: v for k, v in pretrained_dict.items() if (k in model_dict and 'fc' not in k)} #...
www.dbjr.com.cn/article/2142...htm 2025-6-3