添加與訓練模型，以densenet121爲例

添加預訓練模型的方法，自己記錄 先把模型中的pretrained設置爲True,如下     以densenet爲例，在if下面添加 model_dict = model.state_dict() # 將pretrained_dict裏不屬於model_dict的鍵剔除掉 pretrained_dict = {k: v for k, v in model_dict.items() if k in

>>阅读原文<<