Retracing warning on latest tensorflow version

[ailec-wu]

Hello, thank you for releasing your code for public use. When I run model.fit() I'm encountering a retracing warning. Per this stack overflow link I simply added "i = tf.cast(step, tf.int64)" after line 437 in levenberg_marquardt.py and the warning went away. I considered making a PR but this is a relatively small change.

[fabiodimarco]

Hi @ailec-wu , thank you for reporting the issue.
At line 437 there is no variable step defined. Did you meant "i = tf.cast(i, tf.int64)"?
However, I do get an error on my machine if I add the conversion of i to tf.int64.
What version of tensorflow are you using? Could you provide me an example code to reproduce the warning?