feat: Log total training time in seconds

rrutmann · rrutmann · commit b0c3f3d5298c · 2025-07-29T16:29:22.000+02:00
diff --git a/src/modalities/gym.py b/src/modalities/gym.py
@@ -78,7 +78,7 @@ def run(
         )
         end_time = datetime.now()
         print_rank_0(f"Training done at {end_time}.")
-        print_rank_0(f"Total training time: {end_time - start_time}")
+        print_rank_0(f"Total training time: {(end_time - start_time).total_seconds()} seconds.")
 
     def _run_checkpointing(
         self,

Original file line number	Diff line number	Diff line change
`@@ -78,7 +78,7 @@ def run(`
`78`	`78`	`)`
`79`	`79`	`end_time = datetime.now()`
`80`	`80`	`print_rank_0(f"Training done at {end_time}.")`
`81`		`- print_rank_0(f"Total training time: {end_time - start_time}")`
	`81`	`+ print_rank_0(f"Total training time: {(end_time - start_time).total_seconds()} seconds.")`
`82`	`82`
`83`	`83`	`def _run_checkpointing(`
`84`	`84`	`self,`