1 repo
Awesome GitHub RepositoriesModel Hook Interfaces
Mechanisms for intercepting and extracting intermediate tensor states from neural network layers.
Distinguishing note: Focuses on the registration mechanism for layer-wise state extraction rather than general model debugging.
Explore 1 awesome GitHub repository matching artificial intelligence & ml · Model Hook Interfaces. Refine with filters or upvote what's useful.
Awesome Model Hook Interfaces GitHub Repositories
Find the best repos with AI.We'll search the best matching repositories with AI.
- huggingface/pytorch-image-models
huggingface/pytorch-image-models
36,386View on GitHubThis project is a comprehensive library of state-of-the-art neural network architectures designed for image classification and feature extraction. It provides a complete deep learning training framework that supports distributed execution, allowing users to build, train, and fine-tune vision models using optimized schedulers and pre-configured training recipes. The library distinguishes itself through a modular backbone architecture that treats neural networks as decoupled feature extractors, enabling the retrieval of multi-scale outputs for downstream tasks like object detection and segmenta
"A flexible registration mechanism allows users to intercept and extract intermediate tensor states from specific layers during the forward pass."
Pythonaugmixconvnextdistributed-training
