INNER CODE UNIT · Python
get_average_attention
google/break-a-scene · ptp_utils.py:155
def get_average_attention(self):
average_attention = {
key: [item / self.cur_step for item in self.attention_store[key]]
for key in self.attention_store
}
return average_attention
def reset(self):
super(AttentionStore, self).reset()
self.step_store = self.get_empty_store()
self.attention_store = {}
def __init__(self):
super(AttentionStore, self).__init__()
self.step_store = self.get_empty_store()
self.attention_store = {}