【Tensorflow2.0】七、全流程model.fit模型训练方法

文章目录 第一种方法直接用keras的fit方法,以fashion mnist为例 配置超参数 选择指定显卡及自动调用显存 准备数据 使用tf.data来准备训练集和测试集 准备模型定义 开始定义模型,用functional方法 打印模型结构 画出模型结构图并保存 定义优化算法和损失函数 编译模型 定义callbacks 一、生成ckpt每一个epoch训练完保存一个模型 二、当模型验证集精度长时
相关文章
相关标签/搜索