tensorflow.python.framework.errors_impl.InvalidArgumentError: 2 root error(s) found.

tensorflow.python.framework.errors_impl.InvalidArgumentError: 2 root error(s) found.
  (0) Invalid argument: Expected image (JPEG, PNG, or GIF), got unknown format starting with 'gdata/test_set/v'
         [[{{node cond/else/_1/cond/DecodePng}}]]
         [[IteratorGetNext_1]]
         [[IteratorGetNext_1/_137]]
  (1) Invalid argument: Expected image (JPEG, PNG, or GIF), got unknown format starting with 'gdata/test_set/v'
         [[{{node cond/else/_1/cond/DecodePng}}]]
         [[IteratorGetNext_1]]
0 successful operations.
0 derived errors ignored.

训练阶段没问题，问题出现在验证数据集部分，大概率是由于内存问题引起的，解决方式：

在训练文件前加入以下代码

import tensorflow as tf
config = tf.compat.v1.ConfigProto()
config.gpu_options.allow_growth = True
#TensorFlow按需分配显存
config.allow_soft_placement = True
config.log_device_placement = True
config.gpu_options.per_process_gpu_memory_fraction = 0.9
#指定显存分配比例

我尝试了上面的解决代码后，在迭代到10000次时还是出现相同错误，所以可能不是内存分配问题，或许时验证数据集不符合要求导致的，尝试一下，完美解决问题，由于验证数据集不符合要求所以导致错误，使用符合要求的验证集就可以了。