lint fix

pytorch · Feb 26, 2024 · 23b7739 · 23b7739
1 parent b3cb640
commit 23b7739
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train.py b/train.py
@@ -216,7 +216,7 @@ def main(job_config: JobConfig):
             end_timer.record()
             torch.cuda.synchronize()
 
-            curr_iter_time = round(start_timer.elapsed_time(end_timer) * 1e-3,4)
+            curr_iter_time = round(start_timer.elapsed_time(end_timer) * 1e-3, 4)
             train_state.iter_times.append(curr_iter_time)
 
             # if profiler is active