Fix action embedder MLP initialization after FSDP checkpoint load - #5

Fix action embedder MLP initialization after FSDP checkpoint load#5

jjomier merged 2 commits into

mainNVIDIA-Medtech/Cosmos-H-Surgical-Simulator:mainfrom

fix/action-embedder-mlp-reinitNVIDIA-Medtech/Cosmos-H-Surgical-Simulator:fix/action-embedder-mlp-reinitCopy head branch name to clipboard