ray.train.lightning.RayFSDPStrategy.lightning_module_state_dict#

RayFSDPStrategy.lightning_module_state_dict() Dict[str, Any][source]#

Gathers the full state dict to rank 0 on CPU.