time_idx defined but never updated

[nityanandmathur]

Hi @daemon! I am trying to visualize attention maps for each intermediate timestep. For which I tried to change the self.time_idx parameter. Unfortunately, it didn't bring out any change.

daam/daam/trace.py

Line 38 in c30493e

self.time_idx = 0

Looks like this variable is only used in function time_callback().

Could you please let me know how to visualize attention in all intermediate steps?

Thank you!