From 5bf26dadcaa098c1c4c4a0b10f7e0b9fe44a88f8 Mon Sep 17 00:00:00 2001 From: "github-actions[bot]" <41898282+github-actions[bot]@users.noreply.github.com> Date: Wed, 17 May 2023 00:02:26 +0800 Subject: [PATCH] Format code (#296) Co-authored-by: github-actions[bot] --- train_nsf_sim_cache_sid_load_pretrain.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/train_nsf_sim_cache_sid_load_pretrain.py b/train_nsf_sim_cache_sid_load_pretrain.py index f8f40ee..910f576 100644 --- a/train_nsf_sim_cache_sid_load_pretrain.py +++ b/train_nsf_sim_cache_sid_load_pretrain.py @@ -51,10 +51,10 @@ from process_ckpt import savee global_step = 0 + class EpochRecorder: def __init__(self): self.last_time = ttime() - def record(self): now_time = ttime() @@ -64,6 +64,7 @@ class EpochRecorder: current_time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S") return f"[{current_time}] | ({elapsed_time_str})" + def main(): # n_gpus = torch.cuda.device_count() os.environ["MASTER_ADDR"] = "localhost"