深度学习训练集是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万数据集全部训练完成。
真诚希望各位大神解疑。
图片描述

阅读 4.4k
1 个回答
✓ 已被采纳新手上路,请多包涵

问题已经解决,在srez_main.py中,将源代码的
tf.app.flags.DEFINE_integer('train_time', 20,

                        "Time in minutes to train the model")

中的20改为你要训练的时间,大概计算一下完成所需的训练时间,单位为分钟。

撰写回答
你尚未登录,登录后可以
  • 和开发者交流问题的细节
  • 关注并接收问题和回答的更新提醒
  • 参与内容的编辑和改进,让解决方法与时俱进
推荐问题