Frank Lee 1ad3a636b1 [test] fixed torchrec model test (#3167) ...

* [test] fixed torchrec model test

* polish code

* polish code

* polish code

* polish code

* polish code

* polish code

2023-03-20 11:40:25 +08:00

..

update

[test] fixed torchrec model test (#3167)

2023-03-20 11:40:25 +08:00

test_gemini_manager.py

[gemini] add GeminiMemoryManger (#832)

2022-04-24 13:08:48 +08:00

test_param_op.py

[test] align model name with the file name. (#2045)

2022-11-30 15:45:26 +08:00

test_runtime_mem_tracer.py

[gemini] get the param visited order during runtime (#2108)

2022-12-09 16:13:03 +08:00