Hi, I trained my custom data set on yolo.
The training process is successful but when I want to convert checkpoint into a .pb file, this error is appearing:
ValueError: Dimension 1 in both shapes must be equal, but are 6 and 7. Shapes are [?,6,6] and [?,7,7]. for 'concat_1' (op: 'ConcatV2') with input shapes: [?,6,6,256], [?,7,7,1024], [] and with computed input tensors: input[2] = <3>.
Have you any idea how to fix it?
Hi, I trained my custom data set on yolo. The training process is successful but when I want to convert checkpoint into a .pb file, this error is appearing:
ValueError: Dimension 1 in both shapes must be equal, but are 6 and 7. Shapes are [?,6,6] and [?,7,7]. for 'concat_1' (op: 'ConcatV2') with input shapes: [?,6,6,256], [?,7,7,1024], [] and with computed input tensors: input[2] = <3>.
Have you any idea how to fix it?