معرفی شرکت ها


evnn-pytorch-0.1.0


Card image cap
تبلیغات ما

مشتریان به طور فزاینده ای آنلاین هستند. تبلیغات می تواند به آنها کمک کند تا کسب و کار شما را پیدا کنند.

مشاهده بیشتر
Card image cap
تبلیغات ما

مشتریان به طور فزاینده ای آنلاین هستند. تبلیغات می تواند به آنها کمک کند تا کسب و کار شما را پیدا کنند.

مشاهده بیشتر
Card image cap
تبلیغات ما

مشتریان به طور فزاینده ای آنلاین هستند. تبلیغات می تواند به آنها کمک کند تا کسب و کار شما را پیدا کنند.

مشاهده بیشتر
Card image cap
تبلیغات ما

مشتریان به طور فزاینده ای آنلاین هستند. تبلیغات می تواند به آنها کمک کند تا کسب و کار شما را پیدا کنند.

مشاهده بیشتر
Card image cap
تبلیغات ما

مشتریان به طور فزاینده ای آنلاین هستند. تبلیغات می تواند به آنها کمک کند تا کسب و کار شما را پیدا کنند.

مشاهده بیشتر

توضیحات

EVNN: a torch extension for custom event based RNN models.
ویژگی مقدار
سیستم عامل -
نام فایل evnn-pytorch-0.1.0
نام evnn-pytorch
نسخه کتابخانه 0.1.0
نگهدارنده []
ایمیل نگهدارنده []
نویسنده TUD and RUB
ایمیل نویسنده khaleelulla.khan_nazeer@tu-dresden.de
آدرس صفحه اصلی https://tu-dresden.de/ing/elektrotechnik/iee/hpsn
آدرس اینترنتی https://pypi.org/project/evnn-pytorch/
مجوز Apache 2.0
# EvNN: Event-based Neural Networks EvNN is a CUDA and C++ implementation of event-based RNN layers with built-in [DropConnect](http://proceedings.mlr.press/v28/wan13.html) and [Zoneout](https://arxiv.org/abs/1606.01305) regularization. These layers are exposed through C++ and Pytorch APIs for easy integration into your own projects or machine learning frameworks. The code framework and base layers are adopted from [Haste](https://github.com/lmnt-com/haste/) Library. ## EGRU: Event-based Gated Recurrent Unit Event based GRU was publised as a conference paper at ICLR 2023: [Efficient recurrent architectures through activity sparsity and sparse back-propagation through time](https://openreview.net/pdf?id=lJdOlWg8td) (**notable-top-25%**) ![EGRU illustration](./media/images/egru-overview.png) *Illustration of EGRU. **A.** A single unit of the original GRU model adapted from [Cho et al.](#references). **B:** EGRU unit with event generating mechanism. **C:** Heaviside function and surrogate gradient. **D:** Forward state dynamics for two EGRU units (*$i$ *and* $j$ *). **E:** Activity-sparse backward dynamics for two EGRU units (*$i$ *and* $j$ *). (Note that we only have to backpropagate through units that were active or whose state was close to the threshold at each time step.)* ![EvNN Animation](./media/videos/anim/1080p60/EvNNPlot_ManimCE_v0.17.2.gif) Which RNN types are currently supported? - [GRU](https://en.wikipedia.org/wiki/Gated_recurrent_unit) What's included in this project? - a PyTorch API (`evnn_pytorch`) for event based neural networks ## Install Here's what you'll need to get started: - a [CUDA Compute Capability](https://developer.nvidia.com/cuda-gpus) 3.7+ GPU (required only if using GPU) - [CUDA Toolkit](https://developer.nvidia.com/cuda-toolkit) 10.0+ (required only if using GPU) - [PyTorch](https://pytorch.org) 1.3+ for PyTorch integration (GPU optional) - [BLAS](https://netlib.org/blas/) or any BLAS-like library for CPU computation. - [Eigen 3](http://eigen.tuxfamily.org/) to build the C++ examples (optional) Once you have the prerequisites, you can install with pip or by building the source code. <!-- ### Using pip ``` pip install evnn_pytorch ``` --> ### Building from source > **Note** > > Currenty supported only on Linux, use Docker for building on Windows. ```bash make evnn_pytorch # Build PyTorch API ``` If you built the PyTorch API, install it with `pip`: ```bash pip install evnn_pytorch-*.whl ``` If the CUDA Toolkit that you're building against is not in `/usr/local/cuda`, you must specify the `$CUDA_HOME` environment variable before running make: ```bash CUDA_HOME=/usr/local/cuda-10.2 make ``` ## Performance Code for the experiments and benchmarks presented in the paper are published in ``benchmarks`` directory. Note that these benchmarks have additional dependencies as documented in `benchmarks/requirements.txt` ## Documentation ### PyTorch API ```python import torch import evnn_pytorch as evnn # setting use_custom_cuda=False makes the model use pytorch code instead of EvNN extension egru_layer = evnn.EGRU(input_size, hidden_size, zoneout=0.0, batch_first=True, use_custom_cuda=True) egru_layer.cuda() # `x` is a CUDA tensor with shape [N,T,C] x = torch.rand([5, 25, 128]).cuda() y, state = egru_layer(x) ``` The PyTorch API is documented in [`docs/pytorch/evnn_pytorch.md`](docs/pytorch/evnn_pytorch.md). ## Code layout - [`docs/pytorch/`](docs/pytorch): API reference documentation for `evnn_pytorch` - [`frameworks/pytorch/`](frameworks/pytorch): PyTorch API and custom op code - [`lib/`](lib): CUDA kernels and C++ API - [`validation/`](validation): scripts and tests to validate output and gradients of RNN layers - [`benchmarks/`](benchmarks): Experiments from ICLR 2023 Paper. ## Testing use python unittest with this command - Numpy is required for testing ``` python -m unittest discover -p '*_test.py' -s validation ``` > **Note** > > Tests will fail if you set the the dimensions (batch_size,time_steps,input_size, hidden_size) too high, this is because floating point errors can accumulate and cause the units to generate events one timestep off. This causes the numerical tests to fail but the Neural Network training will work without any issues. ## Implementation notes - the EGRU is based on Haste GRU which is in turn based on `1406.1078v1` (same as cuDNN) rather than `1406.1078v3` ## References 1. Nanavati, Sharvil, ‘Haste: A Fast, Simple, and Open RNN Library’, 2020 <https://github.com/lmnt-com/haste/> 1. K. Cho, B. van Merriënboer, C. Gulcehre, D. Bahdanau, F. Bougares, H. Schwenk, and Y. Bengio. Learning phrase representations using RNN encoder–decoder for statistical machine translation. In Proceedings of the 2014 Conference on Empirical Methods in Natural Language Processing (EMNLP), pages 1724–1734, Doha, Qatar, Oct. 2014. Association for Computational Linguistics. doi: 10.3115/v1/D14-1179. [URL](https://aclanthology.org/D14-1179) <!-- if we have tables from the paper here, then add references --> ## Citing this work To cite this work, please use the following BibTeX entry: ``` @inproceedings{ evnn2023, title={Efficient recurrent architectures through activity sparsity and sparse back-propagation through time}, author={Anand Subramoney and Khaleelulla Khan Nazeer and Mark Sch{\"o}ne and Christian Mayr and David Kappel}, booktitle={The Eleventh International Conference on Learning Representations }, year={2023}, url={https://openreview.net/forum?id=lJdOlWg8td}, howpublished={https://github.com/KhaleelKhan/EvNN/} } ``` ## License [Apache 2.0](LICENSE)


نحوه نصب


نصب پکیج whl evnn-pytorch-0.1.0:

    pip install evnn-pytorch-0.1.0.whl


نصب پکیج tar.gz evnn-pytorch-0.1.0:

    pip install evnn-pytorch-0.1.0.tar.gz