fix log error (#144)

Ye-Tian-Zero · facebook-github-bot · commit aa9ebfd44e96 · 2023-06-27T20:59:48.000-07:00
Summary: just a small fix Pull Request resolved: #144 Reviewed By: daniellepintz Differential Revision: D47081346 Pulled By: ananthsub fbshipit-source-id: 711f734844cae77b424d6c9949f6286f2b0185d7
diff --git a/torchsnapshot/scheduler.py b/torchsnapshot/scheduler.py
@@ -49,7 +49,7 @@ def get_process_memory_budget_bytes(pg: PGWrapper) -> int:
                 os.environ["TORCHSNAPSHOT_PER_RANK_MEMORY_BUDGET_BYTES"]
             )
             logger.info(
-                "Manually set process memory budget to {memory_budget_bytes} bytes."
+                f"Manually set process memory budget to {memory_budget_bytes} bytes."
             )
             return memory_budget_bytes
         except Exception as e:

Original file line number	Diff line number	Diff line change
`@@ -49,7 +49,7 @@ def get_process_memory_budget_bytes(pg: PGWrapper) -> int:`
`49`	`49`	`os.environ["TORCHSNAPSHOT_PER_RANK_MEMORY_BUDGET_BYTES"]`
`50`	`50`	`)`
`51`	`51`	`logger.info(`
`52`		`- "Manually set process memory budget to {memory_budget_bytes} bytes."`
	`52`	`+ f"Manually set process memory budget to {memory_budget_bytes} bytes."`
`53`	`53`	`)`
`54`	`54`	`return memory_budget_bytes`
`55`	`55`	`except Exception as e:`