Add train test for cmf_10x, ads_dhen_5x, umia_v1_fbr and blue_reels_vdd_v2 (#1209)
Summary:
Pull Request resolved: https://github.com/pytorch/benchmark/pull/1209
Add train tests for cmf_10x, ads_dhen_5x, umia_v1_fbr and blue_reels_vdd_v2 models.
Upstream code changes break blue_reels_vdd_v2 and umia_v1_fbr. Fix the model code accordingly.
Reviewed By: brad-mengchi
Differential Revision: D39712630
fbshipit-source-id: 07d4e77abf3c331820569ee6e9bb88e703b93904