Unverified Commit b0048b28 authored by girifb's avatar girifb Committed by GitHub
Browse files

Fixing logging, changing info to debug to avoid clutter (#622)



* Changing FSDP init to by pass pg validation for freshly minted pgs inside of init.

* Addressing Min's review comments.

* Changing logging in init to debug from info

* Changing logging in init to debug from info
Co-authored-by: default avatarGiri Anantharaman <giriman@devfair0439.h2.fair>
parent f768eb93
......@@ -271,7 +271,7 @@ class FullyShardedDataParallel(nn.Module):
self._return_full_state_dict = True
init_end = time.time()
logging.info(
logging.debug(
f"FSDP.__init__(done): total_init_time: {(init_end - init_start): .4f} num_params: {(sum(p.numel() for p in self.params))}"
)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment