logging

facebookresearch · Apr 30, 2024 · f787532 · f787532
1 parent 9a22628
commit f787532
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/fairscale/nn/misc/flatten_params_wrapper.py b/fairscale/nn/misc/flatten_params_wrapper.py
@@ -385,6 +385,7 @@ def _unflatten_params_as_views(self) -> None:
         for (_, m, n), p in zip(self._param_infos, ps):
             setattr(p, '_fsdp_weight', True)
             setattr(m, n, p)  # This will set as plain attr
+            #logger.info(f"CHRISLOG: {n=}, {p.requires_grad=}")
             param_views.append(p)
 
         # Save param views for easy access if anyone still wants to access