Open nityanandmathur opened 4 months ago
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.
self.time_idx
https://github.com/castorini/daam/blob/c30493ed0154bfccb6c342400f25cc24599bb1ff/daam/trace.py#L38
Looks like this variable is only used in function time_callback().
time_callback()
Could you please let me know how to visualize attention in all intermediate steps?
Thank you!
Hi @daemon! I am trying to visualize attention maps for each intermediate timestep. For which I tried to change the
self.time_idxparameter. 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!