Train Arch Save

cnn+rnn+attention: vgg(vgg16,vgg19)+rnn(LSTM, GRU)+attention, resnet(resnet_v2_50,resnet_v2_101,resnet_v2_152)+rnnrnn(LSTM, GRU)+attention, inception_v4+rnn(LSTM, GRU)+attention, inception_resnet_v2+rnn(LSTM, GRU)+attention,..... vgg(vgg16,vgg19), resnet(resnet_v2_50,resnet_v2_101,resnet_v2_152), inception_v4, inception_resnet_v2,.....

Project README

此repo是tensorflow版本的,pytorch版本:https://github.com/MachineLP/PyTorch_image_classifier

tensorflow搭建的一个训练框架,包含模型有:vgg(vgg16,vgg19), resnet(resnet_v2_50,resnet_v2_101,resnet_v2_152), inception_v4, inception_resnet_v2等。

此框架主要针对分类任务, 后面会陆续搭建多任务多标签、检测、以及rnn等框架,欢迎关注。 搭建时使用的环境为:Python3.5, tensorflow1.4

具体:

实现基础cnn训练,数据读取方式慢。

优化数据读取的方式,学习率加入衰减。

在train_cnn_v0基础上加入rnn。

在train_cnn_v0基础上加入rnn、attention。

使用多GPU训练(默认两块gpu),以上其他框架使用多GPU,只需把train.py替换掉就可以了。

多任务多标签训练及其总结。

GANs训练及其总结。

理论及其代码实践。

MachineLP的日常代码。

对dl感兴趣,还可以关注我的博客,这是我的博客目录:(地址: http://blog.csdn.net/u014365862/article/details/78422372 ) 本文为博主原创文章,未经博主允许不得转载。有问题可以加微信:lp9628(注明CSDN)。

公众号MachineLN,邀请您扫码关注:

image

Open Source Agenda is not affiliated with "Train Arch" Project. README Source: MachineLP/train_arch
Stars
206
Open Issues
0
Last Commit
3 years ago

Open Source Agenda Badge

Open Source Agenda Rating