prunhild

KO | EN

gitlite — search

#python #react #ai #android #rust #deep-learning #docker #macos #awesome #vue #mcp #ios #openai #open-source

(★ 28)

A small library implementing magnitude-based pruning in PyTorch

prunhild 최신버젼 다운로드

최종 버전 다운로드 (.zip)

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# prunhild
*prunhild* is a small library for neural network pruning based on [PyTorch](https://pytorch.org).

I wrote this library for better structuring my code when replicating the experiments from [The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networks](https://arxiv.org/abs/1803.03635) for a university course.

The library currently only implements simple magnitude-based pruning as used in the previously mentioned paper.
Througout the library magnitude-pruning is referred to as cutoff-pruning.

The library offers a simple class CutoffPruner which offers a similar interface as the [Optimizer](https://pytorch.org/docs/stable/optim.html) class from PyTorch.
The Pruner class takes different strategies (Cutoffs) for computing the binary masks (prune-masks) used for pruning the weights.

## Getting Started

### Installation
You can install the library directly from GitHub:
```bash
# Install from GitHub
pip install git+https://github.com/gfrogat/prunhild
```

## Examples
The folder [examples](./examples) contains an example using MNIST on how to use the library.
If you encounter problems or have any questions don't hesitate to open an issue.

- prunhild example on MNIST - [code](./examples/mnist)

prunhild

prunhild 최신버젼 다운로드

오늘의 추천 프로젝트

Was this content helpful?

Recent Feedback

prunhild

prunhild 최신버젼 다운로드

오늘의 추천 프로젝트

Was this content helpful?

Recent Feedback

Download README