daam icon indicating copy to clipboard operation
daam copied to clipboard

time_idx defined but never updated

Open nityanandmathur opened this issue 9 months ago • 0 comments

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.

https://github.com/castorini/daam/blob/c30493ed0154bfccb6c342400f25cc24599bb1ff/daam/trace.py#L38

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!

nityanandmathur avatar May 23 '24 10:05 nityanandmathur