Refactor torchao and tests to use model architectures from torchao.testing.model_architectures

Open jainapurva opened this issue 10 months ago • 1 comments

PR: https://github.com/pytorch/ao/pull/2036, adds standard model architectures to torchao.testing.model_architectures.py. Replace the existing model definitions from torchao and tests to reuse the model definitions from model_architectures.py. If new definitions are found, add them to model_architectures.py

Eg: Replace https://github.com/pytorch/ao/blob/34421b11d708e8153ccfbd1ef862cf7e8ad42c85/test/quantization/test_quant_api.py#L122-L138 to use torchao.testing.model_architectures.ToyLinearModel

Apr 18 '25 18:04 jainapurva

Can I work on this ?

Apr 19 '25 14:04 XueSongTap