深度学习训练集是20万张,为什么执行训练之后,batch只有230左右就完成训练了,怎么才能把20万张图片全部训练完成。
初学者尝试运行一个GitHub项目(https://github.com/david-gpu/...)
找的数据集是Large-scale CelebFaces Attributes (CelebA)Dataset(http://mmlab.ie.cuhk.edu.hk/p...),其中Align&Cropped Images的数据有20万张。
但是每次进行训练都很快结束,每次的batch只有230,280,300。
要怎么做才能把20万数据集全部训练完成。
真诚希望各位大神解疑。
如果你对这篇内容有疑问,欢迎到本站社区发帖提问 参与讨论,获取更多帮助,或者扫码二维码加入 Web 技术交流群。
绑定邮箱获取回复消息
由于您还没有绑定你的真实邮箱,如果其他用户或者作者回复了您的评论,将不能在第一时间通知您!
发布评论
评论(1)
问题已经解决,在srez_main.py中,将源代码的
tf.app.flags.DEFINE_integer('train_time', 20,
中的20改为你要训练的时间,大概计算一下完成所需的训练时间,单位为分钟。