Skip to content

ybm1/Deep_learning

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

44 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Deep_learning

My DL study.

PytorchTftf.keras的学习代码整理.

分别用PytorchTf(1.x)tf.keras实现了一个Demo,该Demo虽然小,但是五脏俱全, 它有下面的特点:

  • 均支持数据输入,模型定义,训练的模块化.
  • 均支持数据多输入和批量训练.

这里的多输入指的是,每一个样本的都特征包括三部分:

第一部分是三维tensor(模拟图片)

第二部分是二维tensor(模拟句子)

第三部分是一维tensor(模拟一般的ML的特征)

Pytorch的例子中,该部分使用了DatasetDataLoader的API.

Tf的例子中,该部分使用了tf.data.DatasetAPI,支持tfrecoder的写入生成,读取和解析.

tf.keras的例子中,该部分可以无缝衔接的使用Tf中的tf.data.DatasetAPI.

  • 均支持单机多卡训练.
  • 均支持模型的保存和加载(增量训练).

Tf的例子中,模型的定义、单机多卡训练、保存和加载中基本使用的 是Tf低级API,而高级APItf.keras中使用.

  • 均支持tensorboard可视化.

Pytorch的完整的项目代码在Pytorch_learn/MyPytorchDemo中.

Tensorflow的完整的项目代码在Tf_learn/MyTfDemo中.

tf.keras的完整的项目代码在Keras_learn/MyKerasDemo中.

参考的一些资料和博客在参考资料中.

About

My DL study.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages