深度学习项目实践:手写数字识别与词向量学习
1. 构建训练循环
在训练模型时,我们要利用模型进行训练并记录学习到的模型参数,这些操作将在 train.py 中完成。以下是详细步骤:
1.1 导入依赖项
import tensorflow as tf
import hy_param
# MLP Model which we defined in previous step
import model
1.2 定义输入变量
# This will feed the raw images
X = model.X
# This will feed the labels associated with the image
Y = model.Y
1.3 创建保存检查点的文件夹
检查点用于捕获学习过程中权重(W)和偏置(b)的值。我们使用 tf.train.Saver() 函数来保存和恢复检查点。
import os
checkpoint_dir = os.path.abspath(os.path.join(hy_param.checkpoint_dir, "checkpoints"))
checkpoint_prefix = os.path.join(checkpoint_dir, "model")
if
超级会员免费看
订阅专栏 解锁全文

被折叠的 条评论
为什么被折叠?



