Changing the notation
2419007
mp_rank_00_model_states.pt
Detected Pickle imports (7)
- "collections.OrderedDict",
- "torch.FloatStorage",
- "torch.HalfStorage",
- "torch.ByteStorage",
- "torch.Size",
- "torch._utils._rebuild_tensor_v2",
- "__builtin__.set"
How to fix it?
861 MB
Changing the notation
zero_pp_rank_0_mp_rank_00_optim_states.pt
Detected Pickle imports (7)
- "deepspeed.runtime.zero.config.ZeroStageEnum",
- "torch.FloatStorage",
- "collections.OrderedDict",
- "deepspeed.utils.tensor_fragment.fragment_address",
- "torch._utils._rebuild_tensor_v2",
- "deepspeed.runtime.fp16.loss_scaler.DynamicLossScaler",
- "torch.float16"
How to fix it?
77.9 MB
Changing the notation
zero_pp_rank_1_mp_rank_00_optim_states.pt
Detected Pickle imports (7)
- "collections.OrderedDict",
- "torch._utils._rebuild_tensor_v2",
- "deepspeed.utils.tensor_fragment.fragment_address",
- "torch.FloatStorage",
- "deepspeed.runtime.zero.config.ZeroStageEnum",
- "torch.float16",
- "deepspeed.runtime.fp16.loss_scaler.DynamicLossScaler"
How to fix it?
77.9 MB
Changing the notation