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"