Exemplo n.º 1
0
 def on_batch_end(self, state):
     if self.post_each_batch:
         try:
             state['metrics'][self.to] = dict_by_key(state, self.from_k)
         except:
             pass
Exemplo n.º 2
0
 def on_epoch_end(self, state):
     if not self.post_each_batch:
         state['metrics'][self.to] = dict_by_key(state, self.from_k)
Exemplo n.º 3
0
 def on_batch_end(self, state):
     state['metrics'][self.to] = dict_by_key(state, self.from_k)