Skip to content

HazyResearch/structured-nets

master
Switch branches/tags

Name already in use

A tag already exists with the provided branch name. Many Git commands accept both tag and branch names, so creating this branch may cause unexpected behavior. Are you sure you want to create this branch?
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
 
 
 
 
 
 
 
 

structured-nets

Code to accompany the paper Learning Compressed Transforms with Low Displacement Rank.

Requirements

Python 3.6+

PyTorch >=0.4.1

Installing CUDA Extensions

Some functions are written in CUDA for speed. To install them:

cd pytorch/structure/hadamard_cuda
python setup.py install

cd pytorch/structure/diag_mult_cuda
python setup.py install

Example Usage

See our PyTorch section.

About

Structured matrices for compressing neural networks

Resources

License

Stars

Watchers

Forks

Packages

No packages published