迁移学习的基本策略 发表于 2022-05-17 更新于 2022-08-27 分类于 深度学习 阅读次数: Valine: 本文字数: 114 阅读时长 ≈ 1 分钟 迁移学习的基本策略 构造数据集 是否需要对数据进行处理 初始化模型 加载原始模型 设置需要冻住的层 是否需要GPU运算 设置迭代器和损失函数 进行训练 加载需要迁移学习的模型 对训练集进行训练 对验证集进行应该验证 保存验证集上准确率最高的模型 保存模型