JaeWooShin

/

gsm8k_ablation_method_randomseed2

Like 0

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

How to fix it?

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

How to fix it?