Set the counter variable to non-trainable.
Set the counter variable to non-trainable.

This should eliminate two of the warnings in [188](https://github.com/tensorflow/recommenders/issues/188).

PiperOrigin-RevId: 348697996

tensorflow/recommenders,tensorflow/recommenders