Searched refs:epoch_key (Results 1 – 2 of 2) sorted by relevance
16 self.record_batch_start(self.epoch_key(epoch, index))18 self.record_forward_start(self.epoch_key(epoch, index))20 self.record_forward_end(self.epoch_key(epoch, index))21 self.record_backward_start(self.epoch_key(epoch, index))23 self.record_backward_end(self.epoch_key(epoch, index))25 self.record_batch_end(self.epoch_key(epoch, index))
207 def epoch_key(self, epoch, index): member in DdpTrainer