xnn.common.models.hub.checkpoint.wrapped_state_dict# xnn.common.models.hub.checkpoint.wrapped_state_dict(model)[source]# State dict in the trainer’s ForceStressOutput layout, on CPU. Parameters: model (Module) Return type: dict[str, Tensor]
