"src/diffusers/models/controlnets/controlnet_sparsectrl.py" did not exist on "d7001400764acb8de5df343bbc4c54479c0e6ebe"
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 numpy as np
import os
......@@ -42,9 +43,12 @@ class Meter(object):
return total_score / n_tasks
class EarlyStopping(object):
def __init__(self, patience=10, filename="es_checkpoint.pth"):
assert not os.path.exists(filename), \
'Filename {} is occupied. Either rename it or delete it.'.format(filename)
def __init__(self, patience=10, filename=None):
if filename is None:
dt = datetime.datetime.now()
filename = 'early_stop_{}_{:02d}-{:02d}-{:02d}.pth'.format(
dt.date(), dt.hour, dt.minute, dt.second)
self.patience = patience
self.counter = 0
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