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"
        )