老师 我用faster-rcnn训练voc2007数据集时候报这个错误 代码太大发不上来
麻烦老师看一下 谢谢您
InvalidArgumentError (see above for traceback): ValueError: attempt to get argmax of an empty sequence
[[Node: vgg_16/anchor/PyFunc = PyFunc[Tin=[DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_INT32, DT_FLOAT, DT_INT32], Tout=[DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT], token="pyfunc_2", _device="/job:localhost/replica:0/task:0/device:CPU:0"](vgg_16/rpn_cls_score/BiasAdd/_281, _arg_Placeholder_2_0_2, _arg_Placeholder_1_0_1, vgg_16/rois/PyFunc/input_4, vgg_16/ANCHOR_default/generate_anchors, vgg_16/rois/PyFunc/input_6)]]

ps:我显卡是GTX 1050的 是不是显存的原因 我在原有训练代码里面加上了限制gpu的代码 但是不加就溢出了
我加的代码是这个:
config = tf.ConfigProto()
config.gpu_options.allow_growth = config.gpu_options.per_process_gpu_memory_fraction = tf.Session(=config) sess: