EigenDamage-Pytorch
by alecwangcq
Code for "EigenDamage: Structured Pruning in the Kronecker-Factored Eigenbasis" https://arxiv.org/abs/1905.05934
AI summary
Pruning algorithm
A deep learning project implementing structured pruning algorithms in PyTorch for efficient neural network training and inference.
- stars
- 112
- forks
- 19
- watching
- 5
Similar projects
Found by comparing what the projects do, not just their names.
Pruning tool
This project provides a PyTorch implementation of pruning techniques to reduce the computational resources required for neural network inference.
Deep learning toolkit
A Python framework for building deep learning models with optimized encoding layers and batch normalization.
Pruning tool
A tool for training neural networks with pruned weights and evaluating their performance.
Semantic Segmentation Model
An implementation of a deep learning model using PyTorch for semantic segmentation tasks.
Computer Vision Toolkit
A PyTorch toolbox for supporting research and development of domain adaptation, generalization, and semi-supervised learning methods in computer vision.
NLP tutorial
A comprehensive tutorial on deep learning for natural language processing with PyTorch, covering the basics and advancing to linguistic structure prediction.
Image generator
Implementation of a deep learning model for generating high-quality images with improved stability and variation.
Neural network optimizer
An implementation of an optimization algorithm for training neural networks in machine learning environments.
Decoupled Neural Interface
An implementation of synthetic gradients to decouple neural network layers and enable scalable communication between them
Neural net trainer
A PyTorch framework simplifying neural network training with automated boilerplate code and callback utilities
Memory optimizer for deep learning models
A lightweight wrapper around PyTorch to prevent CUDA out-of-memory errors and optimize model execution
Human Segmentation Model
An open source implementation of deep learning-based human segmentation models using PyTorch
A3C algorithm
An implementation of an A3C algorithm for reinforcement learning in Pytorch, with various optimizations and extensions to accelerate training.
A3C algorithm implementation
An implementation of Advantage async Actor-Critic Algorithms in PyTorch for Deep Reinforcement Learning
Optimization algorithm
A PyTorch implementation of an optimization algorithm for continuous control and reinforcement learning tasks
