Open Ianni opened 1 year ago
/cc @chenmoneygithub
/cc @chenmoneygithub
I got this also. I encountered the same error on SWA, then found a collar notebook using MovingAverage before SWA and that has the same error in the collar runtime as I was getting on my laptop.
System information
Describe the bug
Using moving average of optimizers is no longer working and results in
Code to reproduce the issue
Compare e.g. the example notebook from the documentation: https://www.tensorflow.org/addons/tutorials/average_optimizers_callback
Note: if you try it in collab, you have to install tf2.11 first.
Other info / logs
See also https://www.tensorflow.org/addons/api_docs/python/tfa/optimizers/MovingAverage