PyTorch GPT2 model, packaging mechanism, pylint cleanup, model refact… - Githubissues

quic / aimet-model-zoo

Other

296 stars 53 forks source link

PyTorch GPT2 model, packaging mechanism, pylint cleanup, model refact… #28

Closed quic-bharathr closed 1 year ago

quic-bharathr commented 1 year ago

…oring updates

Cleaned up code to minimize errors from static analysis (pylint)
Model refactoring and code updates for 7 models (DeepLabV3, EfficientNetLite0, MobileNetV2, QuickSRNet, FFNet, Inverseform, SSD-MobileNetV2)
Adding package generation capability via cmake and TOML
Releasing first package release of aimet model zoo (installable wheel file binaries)
Added code, documentation and artifacts the PyTorch GPT2 models

Signed-off-by: Bharath Ramaswamy quic_bharathr@quicinc.com