深度学习训练集是20万张,为什么执行训练之后,batch只有230左右就完成训练了,怎么才能把20万张图片全部训练完成。

发布于 2022-09-07 08:03:14 字数 475 浏览 22 评论 0

初学者尝试运行一个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 技术交流群。

扫码二维码加入Web技术交流群

发布评论

需要 登录 才能够评论, 你可以免费 注册 一个本站的账号。

评论(1

云醉月微眠 2022-09-14 08:03:14

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

                        "Time in minutes to train the model")

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

~没有更多了~
我们使用 Cookies 和其他技术来定制您的体验包括您的登录状态等。通过阅读我们的 隐私政策 了解更多相关信息。 单击 接受 或继续使用网站,即表示您同意使用 Cookies 和您的相关数据。
原文