Model save
dbac5ed verified - 240 kB Model save
zero_pp_rank_0_mp_rank_00_optim_states.pt Detected Pickle imports (8)
- "collections.OrderedDict",
- "torch.Tensor",
- "deepspeed.runtime.zero.config.ZeroStageEnum",
- "torch.FloatStorage",
- "deepspeed.runtime.fp16.loss_scaler.DynamicLossScaler",
- "torch._utils._rebuild_tensor_v2",
- "torch._tensor._rebuild_from_type_v2",
- "torch.float16"
How to fix it?
13.9 GB Model save - 240 kB Model save
zero_pp_rank_1_mp_rank_00_optim_states.pt Detected Pickle imports (8)
- "torch.Tensor",
- "deepspeed.runtime.fp16.loss_scaler.DynamicLossScaler",
- "torch._utils._rebuild_tensor_v2",
- "torch.float16",
- "torch.FloatStorage",
- "collections.OrderedDict",
- "torch._tensor._rebuild_from_type_v2",
- "deepspeed.runtime.zero.config.ZeroStageEnum"
How to fix it?
13.9 GB Model save - 240 kB Model save
zero_pp_rank_2_mp_rank_00_optim_states.pt Detected Pickle imports (8)
- "torch.Tensor",
- "deepspeed.runtime.fp16.loss_scaler.DynamicLossScaler",
- "torch._utils._rebuild_tensor_v2",
- "torch.float16",
- "torch.FloatStorage",
- "collections.OrderedDict",
- "torch._tensor._rebuild_from_type_v2",
- "deepspeed.runtime.zero.config.ZeroStageEnum"
How to fix it?
13.9 GB Model save - 240 kB Model save
zero_pp_rank_3_mp_rank_00_optim_states.pt Detected Pickle imports (8)
- "collections.OrderedDict",
- "torch.Tensor",
- "deepspeed.runtime.zero.config.ZeroStageEnum",
- "torch.FloatStorage",
- "deepspeed.runtime.fp16.loss_scaler.DynamicLossScaler",
- "torch._utils._rebuild_tensor_v2",
- "torch._tensor._rebuild_from_type_v2",
- "torch.float16"
How to fix it?
13.9 GB Model save - 240 kB Model save
zero_pp_rank_4_mp_rank_00_optim_states.pt Detected Pickle imports (8)
- "torch.Tensor",
- "deepspeed.runtime.fp16.loss_scaler.DynamicLossScaler",
- "torch._utils._rebuild_tensor_v2",
- "torch.float16",
- "torch.FloatStorage",
- "collections.OrderedDict",
- "torch._tensor._rebuild_from_type_v2",
- "deepspeed.runtime.zero.config.ZeroStageEnum"
How to fix it?
13.9 GB Model save - 240 kB Model save
zero_pp_rank_5_mp_rank_00_optim_states.pt Detected Pickle imports (8)
- "collections.OrderedDict",
- "torch.Tensor",
- "deepspeed.runtime.zero.config.ZeroStageEnum",
- "torch.FloatStorage",
- "deepspeed.runtime.fp16.loss_scaler.DynamicLossScaler",
- "torch._utils._rebuild_tensor_v2",
- "torch._tensor._rebuild_from_type_v2",
- "torch.float16"
How to fix it?
13.9 GB Model save - 240 kB Model save
zero_pp_rank_6_mp_rank_00_optim_states.pt Detected Pickle imports (8)
- "collections.OrderedDict",
- "deepspeed.runtime.fp16.loss_scaler.DynamicLossScaler",
- "torch._utils._rebuild_tensor_v2",
- "torch.Tensor",
- "torch._tensor._rebuild_from_type_v2",
- "torch.float16",
- "deepspeed.runtime.zero.config.ZeroStageEnum",
- "torch.FloatStorage"
How to fix it?
13.9 GB Model save - 240 kB Model save
zero_pp_rank_7_mp_rank_00_optim_states.pt Detected Pickle imports (8)
- "collections.OrderedDict",
- "torch.Tensor",
- "deepspeed.runtime.zero.config.ZeroStageEnum",
- "torch.FloatStorage",
- "deepspeed.runtime.fp16.loss_scaler.DynamicLossScaler",
- "torch._utils._rebuild_tensor_v2",
- "torch._tensor._rebuild_from_type_v2",
- "torch.float16"
How to fix it?
13.9 GB Model save
