search

ShishirPatil / poet

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

Transforming Models from PyTorch/Tensorflow to POET #9

Open arnavsinghvi11 opened 1 year ago

arnavsinghvi11 commented 1 year ago

This PR is the first step towards translating models defined in Torch to Tensorflow transparently. We define network transformation functions which are then composed together for converting PyTorch and Tensorflow network layers to POET computation layers.

Tested for ResNet18, ResNet50, and VGG16.

arnavsinghvi11 commented 1 year ago

created network transformation function for converting TensorFlow network layers to POET computation layers

testing is done for ResNet model

arnavsinghvi11 commented 1 year ago

created network transformation function for converting PyTorch model graph to graph with POET computation layer nodes

testing is done for ResNet18, ResNet50