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,邀请您扫码关注: