【发布时间】:2017-10-24 04:43:48
【问题描述】:
我正在使用带有 GTX 1060 GPU 的 TensorFlow Inception v3 实现图像分类。我在 9 个不同的分类器中训练了大约 3000 张图像,其中每个分类器包含 100 到 500 张图像(100*100 像素),使用诗人的张量。 但是当我有超过 500 个图像时,它会抛出错误并且训练失败。具有较少图像的同一类训练没有错误。我每个班级的图像数据大小约为 5000~10000,但只有我可以训练大约 300~400 的数据大小。更多的数据,它会给出以下 GPU 错误。
规格如下:
- Windows 7 与 GeForce GTX 1060
- Anaconda 3 与 python 3.5
- Tensorflow Inception V3
【问题讨论】:
标签: python tensorflow deep-learning image-recognition tensorflow-gpu