Commit 5636e29
refactor the models in test_model.py
Summary:
# context
* refactor the models in test_utils.test_models
* add docstring
* absorb optional arguments into `**kwargs`
Differential Revision: D846152541 parent 24bb848 commit 5636e29
File tree
2 files changed
+436
-205
lines changed- torchrec
- distributed/test_utils
- modules
2 files changed
+436
-205
lines changed
0 commit comments