Skip to content

Commit

Permalink
remove logging
Browse files Browse the repository at this point in the history
  • Loading branch information
chrisxcai committed Apr 29, 2024
1 parent d1102ce commit 9a22628
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions fairscale/nn/misc/flatten_params_wrapper.py
Original file line number Diff line number Diff line change
Expand Up @@ -372,13 +372,13 @@ def _unflatten_params_as_views(self) -> None:
self.flat_param unchanged.
"""
assert self.is_flattened
logger.info(f"CHRISLOG: {self._require_backward_grad_sync=}")
#logger.info(f"CHRISLOG: {self._require_backward_grad_sync=}")
if self._require_backward_grad_sync:
logger.info("CHRISLOG: calling self.get_param_views() without torch.no_grad()")
#logger.info("CHRISLOG: calling self.get_param_views() without torch.no_grad()")
ps = self.get_param_views()
else:
with torch.no_grad():
logger.info("CHRISLOG: calling self.get_param_views() with torch.no_grad()")
#logger.info("CHRISLOG: calling self.get_param_views() with torch.no_grad()")
ps = self.get_param_views()

param_views = []
Expand Down

0 comments on commit 9a22628

Please sign in to comment.