def batch_end(self):
    self.batch_time.update((time.time() - self.start))
    self.start = time.time()
def batch_end(self):
    self.batch_time.update((time.time() - self.start))
    self.start = time.time()