import sys import time class Logger(): def __init__(self): self.start_time = time.time() self.epoch_start_time = self.start_time def print_training_log(self, current_epoch, total_epoch, **kargs): current_time = time.time() epoch_time = current_time - self.epoch_start_time total_time = current_time - self.start_time estimated_total_time = total_time * total_epoch / (current_epoch + 1) remaining_time = estimated_total_time - total_time self.epoch_start_time = current_time sys.stdout.write( f"epoch : {current_epoch}/{total_epoch}\n" )