ValueError: Cannot reshape a tensor with 2945760 elements to shape [2,76,76,3,36] (1247616 elements)

[StevenW1999]

When training the model from scratch with 608x608 images i get the following error:

ValueError: Cannot reshape a tensor with 2945760 elements to shape [2,76,76,3,36] (1247616 elements) for '{{node custom_loss/Reshape_1}} = Reshape[T=DT_FLOAT, Tshape=DT_INT32](Yolo_v4/BN_138/FusedBatchNormV3, custom_loss/Reshape_1/shape)' with input shapes: [2,76,76,255], [5] and with input tensors computed as partial shapes: input[1] = [2,76,76,3,36].

I am not sure where it goes wrong, could you help me please? Thank you!