ge0405

Use Torchrec OSS Planner (2/2)

4

Summary: Incorporate Torchrec's OSS planner into train_module. The OSS planner can be enabled by setting "`use_torchrec_oss_planner`" flag to true in SharderOptions (https://fburl.com/code/xjlbrscm). When the flag is true, `run_planner` calls `run_oss_planner`...