[NNX] Checkpoint restore: error on weights missing from the checkpoint#4342

ecnal-cienet wants to merge 1 commit into

mainAI-Hypercomputer/maxtext:mainfrom

feat/nnx-ckpt-missing-param-policyAI-Hypercomputer/maxtext:feat/nnx-ckpt-missing-param-policyCopy head branch name to clipboard