我正在尝试新发布的用于 tensorflow 的对象检测 API。我使用了教程中的示例训练程序,即在宠物数据集上微调FRCNN-Resnet模型。仅使用一个 GPU,当我将批量大小更改为大于 1 的值(默认值为 1)时,总是会显示一条错误消息。错误消息如下所示:
InvalidArgumentError(参见上文的回溯):ConcatOp:输入的尺寸应匹配:shape[0] = [1,750,600,3] 与 shape[1] = [1,600,804,3]
我正在尝试新发布的用于 tensorflow 的对象检测 API。我使用了教程中的示例训练程序,即在宠物数据集上微调FRCNN-Resnet模型。仅使用一个 GPU,当我将批量大小更改为大于 1 的值(默认值为 1)时,总是会显示一条错误消息。错误消息如下所示:
InvalidArgumentError(参见上文的回溯):ConcatOp:输入的尺寸应匹配:shape[0] = [1,750,600,3] 与 shape[1] = [1,600,804,3]