Skip to content

Commit

Permalink
[FSDP][Easy] Remove extraneous print
Browse files Browse the repository at this point in the history
Pull Request resolved: pytorch#77705

Approved by: https://github.com/zhaojuanmao
  • Loading branch information
awgu authored and pytorchmergebot committed May 18, 2022
1 parent ccc991b commit 93b20b0
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions torch/distributed/fsdp/fully_sharded_data_parallel.py
Original file line number Diff line number Diff line change
Expand Up @@ -3170,8 +3170,6 @@ def _check_rebuild_full_params(self, param: FlatParameter):
curr_iter_param_names = [
eod.get_unflat_param_names(index) for index in curr_param_order
]
print(first_iter_param_names, type(first_iter_param_names))
print(curr_iter_param_names, type(curr_iter_param_names))
warnings.warn(
"Forward order differs from that of the first iteration "
f"on rank {self.rank} -- collectives are unchecked and may "
Expand Down

0 comments on commit 93b20b0

Please sign in to comment.