当前位置: 首页 > 工具软件 > Tensor2Tensor > 使用案例 >

google的tensor2tensor的学习和使用

詹钊
2023-12-01

前言

  • 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.目录结构

因为有需要注意的点,先看下我自定义样本的目录结构:

 类似资料: