ver217
|
6ae8ed0407
|
[lazyinit] add correctness verification (#3147)
* [lazyinit] fix shared module
* [tests] add lazy init test utils
* [tests] add torchvision for lazy init
* [lazyinit] fix pre op fn
* [lazyinit] handle legacy constructor
* [tests] refactor lazy init test models
* [tests] refactor lazy init test utils
* [lazyinit] fix ops don't support meta
* [tests] lazy init test timm models
* [lazyinit] fix set data
* [lazyinit] handle apex layers
* [tests] lazy init test transformers models
* [tests] lazy init test torchaudio models
* [lazyinit] fix import path
* [tests] lazy init test torchrec models
* [tests] update torch version in CI
* [tests] revert torch version in CI
* [tests] skip lazy init test
|
2023-03-17 13:49:04 +08:00 |
|
Frank Lee
|
ed19290560
|
[booster] implemented mixed precision class (#3151)
* [booster] implemented mixed precision class
* polish code
|
2023-03-17 11:00:15 +08:00 |
|
YuliangLiu0306
|
ecd643f1e4
|
[test] add torchrec models to test model zoo (#3139)
|
2023-03-15 05:46:04 +00:00 |
|
ver217
|
14a115000b
|
[tests] model zoo add torchaudio models (#3138)
* [tests] model zoo add torchaudio models
* [tests] refactor torchaudio wavernn
* [tests] refactor fx torchaudio tests
|
2023-03-15 11:51:16 +08:00 |
|
Frank Lee
|
6d48eb0560
|
[test] added transformers models to test model zoo (#3135)
|
2023-03-15 11:26:10 +08:00 |
|
Frank Lee
|
a674c63348
|
[test] added torchvision models to test model zoo (#3132)
* [test] added torchvision models to test model zoo
* polish code
* polish code
* polish code
* polish code
* polish code
* polish code
|
2023-03-15 10:42:07 +08:00 |
|
HELSON
|
1216d1e7bd
|
[tests] diffuser models in model zoo (#3136)
* [tests] diffuser models in model zoo
* remove useless code
* [tests] add diffusers to requirement-test
|
2023-03-14 17:20:28 +08:00 |
|
Frank Lee
|
86ac782d7c
|
[test] added timm models to test model zoo (#3129)
* [test] added timm models to test model zoo
* polish code
* polish code
* polish code
* polish code
* polish code
|
2023-03-14 14:29:18 +08:00 |