TAPIR loss computation clarification

[shivanimall]

Hello,

I saw in a comment in tapir_model.py code that says: At training time there's only one iteration, so we'll just get the final refined output. However, in the paper, in section 3.2 it says At each iteration, we use the same loss L on the output, weighting each iteration the same as the initialization. I wanted to clarify whether or not the loss is computed per refinement iteration (i.e, 5 (num_pips_iters: 4 + init: 1))?

Thank you!