4、深度学习项目实践:手写数字识别与词向量学习

深度学习项目实践:手写数字识别与词向量学习

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
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值