step 3200
a38a5b8 verified - 1.86 kB step 3200
opt_state.pkl Detected Pickle imports (11)
- "ml_dtypes.bfloat16",
- "optax._src.base.EmptyState",
- "optax._src.transform.ScaleByScheduleState",
- "numpy._core.multiarray._reconstruct",
- "numpy.dtype",
- "optax.transforms._masking.MaskedNode",
- "optax._src.transform.ScaleByAdamState",
- "optax.contrib._muon.MuonState",
- "optax.transforms._masking.MaskedState",
- "numpy.ndarray",
- "optax.transforms._combining.PartitionState"
How to fix it?
3.15 GB step 3200 params.pkl Detected Pickle imports (4)
- "numpy.dtype",
- "numpy._core.multiarray._reconstruct",
- "numpy.ndarray",
- "ml_dtypes.bfloat16"
How to fix it?
3.15 GB step 3200
