sujr

/

scaling_exp_refSoM

like 0

• "collections.OrderedDict",
• "torch._utils._rebuild_tensor_v2",
• "deepspeed.runtime.fp16.loss_scaler.LossScaler",
• "deepspeed.runtime.zero.config.ZeroStageEnum",
• "torch.FloatStorage",
• "deepspeed.utils.tensor_fragment.fragment_address"

How to fix it?

• "deepspeed.utils.tensor_fragment.fragment_address",
• "torch.FloatStorage",
• "deepspeed.runtime.zero.config.ZeroStageEnum",
• "collections.OrderedDict",
• "torch._utils._rebuild_tensor_v2",
• "deepspeed.runtime.fp16.loss_scaler.LossScaler"

How to fix it?

• "deepspeed.utils.tensor_fragment.fragment_address",
• "torch.FloatStorage",
• "deepspeed.runtime.zero.config.ZeroStageEnum",
• "collections.OrderedDict",
• "torch._utils._rebuild_tensor_v2",
• "deepspeed.runtime.fp16.loss_scaler.LossScaler"

How to fix it?

• "collections.OrderedDict",
• "torch._utils._rebuild_tensor_v2",
• "deepspeed.runtime.fp16.loss_scaler.LossScaler",
• "deepspeed.runtime.zero.config.ZeroStageEnum",
• "torch.FloatStorage",
• "deepspeed.utils.tensor_fragment.fragment_address"

How to fix it?

• "collections.OrderedDict",
• "torch.BFloat16Storage",
• "torch._utils._rebuild_tensor_v2",
• "torch.Size",
• "__builtin__.set"

How to fix it?