Skip to content

Commit

Permalink
Fix logger formatting for partitioning flags (#4728)
Browse files Browse the repository at this point in the history
Logger takes only the first argument as a message.

Co-authored-by: Logan Adams <114770087+loadams@users.noreply.github.com>
  • Loading branch information
OAfzal and loadams authored Nov 29, 2023
1 parent ba6bfd0 commit 8640b8e
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion deepspeed/runtime/pipe/engine.py
Original file line number Diff line number Diff line change
Expand Up @@ -136,7 +136,7 @@ def __init__(self, has_bool_tensors=False, *super_args, **super_kwargs):
assert isinstance(self._config.pipeline['grad_partitioned'], bool)
self.is_pipe_partitioned = self.is_model_parallel and self._config.pipeline['pipe_partitioned']
self.is_grad_partitioned = self.is_model_parallel and self._config.pipeline['grad_partitioned']
logger.info(f'is_pipe_partitioned= {self.is_pipe_partitioned}',
logger.info(f'is_pipe_partitioned= {self.is_pipe_partitioned} '
f'is_grad_partitioned= {self.is_grad_partitioned}')

model_parameters = filter(lambda p: p.requires_grad, self.module.parameters())
Expand Down

0 comments on commit 8640b8e

Please # to comment.