finetune网络,我习惯分两步:首先训练fc层,迭代几个epoch后保存模型;然后基于得到的模型,训练整个网络,一般迭代40-60个epoch可以得到稳定的结果。 total_loss会一直下降的,过程中可以评测下模型在测试集上的表现。真正的loss往往包括两部分。后面total_loss的下降主要是正则项的功劳了。 3.评估模型 1.混淆矩阵必不可...
首先,我们需要加载训练好的深度学习模型: classModel:defload_model(self,model_path):# 加载模型的代码passclassCNNModel(Model):defload_model(self,model_path):# 加载CNN模型的代码passmodel=CNNModel()model.load_model('path_to_model') 1. 2. 3. 4. 5. 6. 7. 8. 9. 10. 11. 12. 接着,我...