Format code (#296)
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
This commit is contained in:
parent
0b0bd911d9
commit
5bf26dadca
|
@ -51,10 +51,10 @@ from process_ckpt import savee
|
||||||
|
|
||||||
global_step = 0
|
global_step = 0
|
||||||
|
|
||||||
|
|
||||||
class EpochRecorder:
|
class EpochRecorder:
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
self.last_time = ttime()
|
self.last_time = ttime()
|
||||||
|
|
||||||
|
|
||||||
def record(self):
|
def record(self):
|
||||||
now_time = ttime()
|
now_time = ttime()
|
||||||
|
@ -64,6 +64,7 @@ class EpochRecorder:
|
||||||
current_time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
current_time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
||||||
return f"[{current_time}] | ({elapsed_time_str})"
|
return f"[{current_time}] | ({elapsed_time_str})"
|
||||||
|
|
||||||
|
|
||||||
def main():
|
def main():
|
||||||
# n_gpus = torch.cuda.device_count()
|
# n_gpus = torch.cuda.device_count()
|
||||||
os.environ["MASTER_ADDR"] = "localhost"
|
os.environ["MASTER_ADDR"] = "localhost"
|
||||||
|
|
Loading…
Reference in New Issue