File name
Commit message
Commit date
2023-06-23
2023-06-26
File name
Commit message
Commit date
2023-06-26
2023-06-26
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"
)