前言
- tensor2tensor(t2t)是google基于tensorflow新开源的深度学习库,该库将深度学习所需要的元素(数据集、模型、学习率、超参数等)封装成标准化的统一接口,在使用其做模型训练时可以更加的灵活。
当前环境
- mac 10.13.3
- tensorflow 1.6.0
- tensor2tensor 1.5.5
安装模块
- 源码下载:https://github.com/tensorflow/tensor2tensor
- sudo pip install tensorflow==1.6.0
- sudo pip install tensor2tensor==1.5.5
开启学习之旅
在下载的源码中有自带的一些简单的测试样例。如mnist,,可以参考https://github.com/tensorflow/tensor2tensor/blob/master/tensor2tensor/notebooks/hello_t2t.ipynb
来做初步的了解。
同时t2t支持自定义的样本数据和自定义的参数配置,下面以我自定义了一个训练样本开始介绍。
已将下述的代码上传:https://download.csdn.net/download/csa121/10672326
根据下述的介绍也可以自己搭建一个环境的。
0.目录结构
因为有需要注意的点,先看下我自定义样本的目录结构: