Skip to content

Commit 5bf26da

Browse files
Format code (#296)
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
1 parent 0b0bd91 commit 5bf26da

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

train_nsf_sim_cache_sid_load_pretrain.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,10 +51,10 @@
5151

5252
global_step = 0
5353

54+
5455
class EpochRecorder:
5556
def __init__(self):
5657
self.last_time = ttime()
57-
5858

5959
def record(self):
6060
now_time = ttime()
@@ -64,6 +64,7 @@ def record(self):
6464
current_time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
6565
return f"[{current_time}] | ({elapsed_time_str})"
6666

67+
6768
def main():
6869
# n_gpus = torch.cuda.device_count()
6970
os.environ["MASTER_ADDR"] = "localhost"

0 commit comments

Comments
 (0)