How to get the number of iterations used to adjust the weight of losses in training.

Hi!
I have now added a factor to my loss and I only need it to work for the first 10000 iterations, how can I get the current number of iterations in the py file of the loss function.
Thanks!