(self)
| 208 | |
| 209 | |
| 210 | def get_average_attention(self): |
| 211 | average_attention = {key: [item / self.cur_step for item in self.attention_store[key]] for key in self.attention_store} |
| 212 | return average_attention |
| 213 | |
| 214 | |
| 215 | def reset(self): |
no outgoing calls
no test coverage detected