Unknown function 'ops.prim.requires_grad'.

lutzroeder / netron

Visualizer for neural network, deep learning and machine learning models

https://netron.app

MIT License

28.15k stars 2.79k forks source link

Unknown function 'ops.prim.requires_grad'. #1328

Closed WorldPeace123 closed 2 months ago

WorldPeace123 commented 2 months ago

Netron app and version: <winget install -s winget netron>
OS and browser version:

Steps to Reproduce:

1.the netron cant look the model(use pytorch , save as scriptmodle),and report the bug of "Unknown function 'ops.prim.requires_grad'.",I dont know how to do.(i find the keywords"ops.prim.requires_grad."in pytorch code and my softwar but cantf find the key words).what can i do that can get the model input shape?thanks ! 微信截图_20240821172741

lutzroeder commented 2 months ago

@WorldPeace123 can you share a model file to reproduce the issue.

WorldPeace123 commented 2 months ago

@lutzroeder Hello my friend?the file is too big ,can you give me a email,that i send to you ?

lutzroeder commented 2 months ago

@WorldPeace123 can you share the file via Dropbox or Google Drive or create a smaller file that produces the same issue. pytorch_strip_weights.zip can be used to remove the weights from a PyTorch file to shrink it in size when compressed.