search

aws-neuron / aws-neuron-sdk

Powering AWS purpose-built machine learning chips. Blazing fast and cost effective, natively integrated into PyTorch and TensorFlow and integrated with your favorite AWS services
https://aws.amazon.com/machine-learning/neuron/
Other
450 stars 152 forks source link

Profiling Unknown framework used #433

Closed alejoGT1202 closed 2 years ago

alejoGT1202 commented 2 years ago

I'm trying to Profile a yoloR converted to Pytorch Neuron. Followed the steps specified here. I'm able to see that the .ntff, .pb and .neff files where after running the inference on a single image.

When I connect to the localhost I can't see the operators in the graph and just display the following graph

image

Also on the terminal where tensorboard is running I get the following message: 

TensorBoard 1.15.0 at http://ip-172-31-16-203:6006/ (Press CTRL+C to quit)
W0616 07:14:16.198482 140492762142464 neuron_plugin.py:289] Unknown framework used - defaulting to TensorFlow supported ops

Should I update TensorBoard and/or the plugin to see the results?

Thanks for the help!

jeffhataws commented 2 years ago

We will take a look. To help us further, please provide versions of the Neuron packages.

pip list | grep neuron yum list | grep neuron # on AL2 apt list | grep neuron # on ubuntu

If you have the DLAMI version that would also help.

Thanks.

jeffhataws commented 2 years ago

Hi alejoGT1202. Thanks for filing the issue. As noted in the overview for TensorBoard Neuron, graph visualization is currently only supported for TensorFlow-Neuron. Support for PyTorch-Neuron visualization will be added in a future release.

aws-joshim commented 2 years ago

@alejoGT1202, please let us know if we can be of any further help for this issue before closing the ticket