Method between_steps

src/attention_controllers.py:25–32

(self)

    

    def between_steps(self):
        # if len(self.attention_store) == 0:
        #     self.attention_store = self.step_store
        # else:
        #     for key in self.attention_store:
        #         for i in range(len(self.attention_store[key])):
        #             self.attention_store[key][i] += self.step_store[key][i]
        self.step_store = self.get_empty_store()

    # 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}