search

ShishirPatil / poet

ML model training for edge devices
https://shishirpatil.github.io/poet/
Apache License 2.0
157 stars 18 forks source link

Pytorch integration: Remat #10

Open arnavsinghvi11 opened 1 year ago

arnavsinghvi11 commented 1 year ago

Implementing rematerialization for PyTorch models. With remat-only we can train ResNet18 on CIFAR-10 with ~93.5% accuracy in 150 epochs.