fix: fixes #264 where tied weights check didn't work on fsdp1#284

Merged

parthchadha merged 3 commits intomainfrom

pchadha/tied-weight-model

Apr 29, 2025

Commits on Apr 28, 2025

fix: fixes #264 where tied weights check didn't work on fsdp1
parthchadha
committed
Update nemo_rl/models/policy/dtensor_policy_worker.py

parthchadha
and
SahilJain314
committed
Update nemo_rl/models/policy/fsdp1_policy_worker.py

parthchadha
and
SahilJain314
committed