Unverified Commit 35bed2a9 authored by Mufei Li's avatar Mufei Li Committed by GitHub
Browse files

Rename early stop (#773)

parent 02029dce
import datetime
import dgl import dgl
import numpy as np import numpy as np
import os import os
...@@ -42,9 +43,12 @@ class Meter(object): ...@@ -42,9 +43,12 @@ class Meter(object):
return total_score / n_tasks return total_score / n_tasks
class EarlyStopping(object): class EarlyStopping(object):
def __init__(self, patience=10, filename="es_checkpoint.pth"): def __init__(self, patience=10, filename=None):
assert not os.path.exists(filename), \ if filename is None:
'Filename {} is occupied. Either rename it or delete it.'.format(filename) dt = datetime.datetime.now()
filename = 'early_stop_{}_{:02d}-{:02d}-{:02d}.pth'.format(
dt.date(), dt.hour, dt.minute, dt.second)
self.patience = patience self.patience = patience
self.counter = 0 self.counter = 0
self.filename = filename self.filename = filename
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment