[feature request] add support for pytorch models in function set_model_graph

[Idodox]

Currently, there's only support for TF model graphs.

[dsblank]

Good idea! Do you have code that turns a pytorch model graph into JSON or a string rep?

[Idodox]

You can get some info by simply printing the model:

from torchvision import models
model = models.vgg16()
print(model)

There's also a package called torchsummary that offers more detail:

from torchvision import models
from torchsummary import summary

vgg = models.vgg16()
summary(vgg, (3, 224, 224))

[dsblank]

That is useful info. Thanks @Idodox , it is on our list!

[george-gca]

There is also the option of converting the model to ONNX.

[dsblank]

Indeed! We have some work on that front coming soon. I'll report back here.

[github-actions[bot]]

This issue is stale because it has been open 30 days with no activity. Remove stale label or comment or this will be closed in 5 days.

[github-actions[bot]]

This issue was closed because it has been stalled for 5 days with no activity.