self.assertEqual(predictions[cnma.BOX_OFFSET][0].shape,
                     (batch_size, height, width // stride, 2))