Skip to content

tosaka2/pruning_chainer

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

pruning_chainer

Pruning implemented in chainer

training and model implementation is from https://github.com/chainer/chainer/tree/master/examples/cifar .

Usage

    # Set pruning
    masks = pruning.create_model_mask(model, args.pruning)
    trainer.extend(pruning.pruned(model, masks))

    # Run the training
    trainer.run()

About

Magunitude-based pruning implemented in Chainer

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages