Skip to content

Conversation

@Shlok-Saxena
Copy link
Contributor

This Pull request

Fixes a shape mismatch in the Keras test model generator used for SOFIE tests and improves reproducibility of generated models.

Changes or fixes:

  • Fixed incorrect training input/target tensor shapes in generateBinaryOpModel() from (2,1) to (2,2) to match the model input signature Input(shape=(2,))
  • Added tf.keras.utils.set_random_seed(0) so model initialization and generated outputs are reproducible across runs, improving stability of SOFIE Keras parser tests.

Checklist:

  • tested changes locally
  • updated the docs (if necessary)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants