迁移学习的基本策略

迁移学习的基本策略

  • 构造数据集
    • 是否需要对数据进行处理
  • 初始化模型
    • 加载原始模型
  • 设置需要冻住的层
  • 是否需要GPU运算
  • 设置迭代器和损失函数
  • 进行训练
    • 加载需要迁移学习的模型
    • 对训练集进行训练
    • 对验证集进行应该验证
    • 保存验证集上准确率最高的模型
    • 保存模型