【MXNet学习23】4.2 训练代码详细解读

4.1节中介绍了一个完整的MNIST手写数字体分类代码,这几乎是每一个入门深度学习的人都会运行的实验。一个比较好的代码阅读习惯既可以是从全局到局部,也可以是从局部到全局,这样才能建立起对代码的完整认识。目前开源的深度学习相关的项目非常多,但不论这些项目的代码量如何,基本上每个项目的代码都是分模块维护的,比如专门的数据读取模块、模型构建模块、参数配置模块等,这对于阅读复杂项目的代码尤其有帮助。因此希望读者能够逐渐养成这样的习惯,并应用在今后的代码书写中。

通过4.1节的介绍,相信你对整个训练代码有了一个直观的认识,接下来就将这份代码分解成几个小模块,包括训练参数配置、数据读取、网络结构搭建和训练模型四大部分,然后分别介绍每个模块的内容与作用。

4.2.1 训练参数配置

在大部分用Python实现的深度学习项目中,argparse模块是非常常见的,因为在Python中,argparse模块扮演着解析命令行参数的角色。这是什么意思呢?前面介绍过这份代码保存在train_mnist_code4-1.py中,因此当你启动训练时,可以直接在train_mnist_code4-1.py文件所在的目录运行以下命令:

$ python train_mnist_code4-1.py

运行命令没有涉及任何参数,这是因为在train_mnist_code4-1.py脚本中定义了一些训练相关的参数并设定了这些参数的默认值,因此直接通过上述命令就能启动训练。但是如果你不想用默认设置的参数来进行训练,比如bat

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

还能坚持

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值