in src/helpers.py [0:0]
def __init__(self, patience: int, filename: Optional[str] = None) -> None:
if filename is None:
# Name checkpoint based on time
dt = datetime.now()
filename = "early_stop_{}_{:02d}-{:02d}-{:02d}.pth".format(
dt.date(), dt.hour, dt.minute, dt.second
)
filename = os.path.join("/opt/ml/model", filename)
self.patience = patience
self.counter = 0
self.filename = filename
self.best_score = None
self.early_stop = False