archinetai / surgeon-pytorch Goto Github PK
View Code? Open in Web Editor NEWA library to inspect and extract intermediate layers of PyTorch models.
License: MIT License
A library to inspect and extract intermediate layers of PyTorch models.
License: MIT License
Is it possible to save the results from the backbone and apply them on the heads of the all the other models. My goal was to try to save time by avoiding repeating the backbone part. Instead of running the 3 complete models (left), only run the backbone 1 time and switch only the heads for the 3 models (right), therefore not repeating executing the backbone every time in yolov5 model.
Thank you for the help!
Hi, I noticed that the current version does not support parallel models (at least those created using torch.nn.DataParallel
) since the forward hook does not differentiate between the different copies of the model and a model wrapped with Inspect
will just return the intermediate features of the last copy of the parallelized model to run.
Are you planning on fixing this issue/supporting this use case?
Hi, I'm a student studying the deep learning compiler for DL frameworks.
I'm also making a program that handles a computational graph(CG) of pytorch by using the 'torch.fx' library, but always failed to derive a output result of node constituting a CG.
I wonder how you make it possible to read the output result of a node with the Extract class?
Is it related to the override function name of 'call_module'?
Thank you for reading my quetion.
A declarative, efficient, and flexible JavaScript library for building user interfaces.
๐ Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
An Open Source Machine Learning Framework for Everyone
The Web framework for perfectionists with deadlines.
A PHP framework for web artisans
Bring data to life with SVG, Canvas and HTML. ๐๐๐
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
Some thing interesting about web. New door for the world.
A server is a program made to process requests and deliver data to clients.
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
Some thing interesting about visualization, use data art
Some thing interesting about game, make everyone happy.
We are working to build community through open source technology. NB: members must have two-factor auth.
Open source projects and samples from Microsoft.
Google โค๏ธ Open Source for everyone.
Alibaba Open Source for everyone
Data-Driven Documents codes.
China tencent open source team.