returnn

The RWTH extensible training framework for universal recurrent neural networks

Github星跟踪图

==================
Welcome to RETURNN

GitHub repository <https://github.com/rwth-i6/returnn>__.
RETURNN paper 2016 <https://arxiv.org/abs/1608.00895>,
RETURNN paper 2018 <https://arxiv.org/abs/1805.05225>
.

RETURNN - RWTH extensible training framework for universal recurrent neural networks,
is a Theano/TensorFlow-based implementation of modern recurrent neural network architectures.
It is optimized for fast and reliable training of recurrent neural networks in a multi-GPU environment.

Features include:

  • Mini-batch training of feed-forward neural networks
  • Sequence-chunking based batch training for recurrent neural networks
  • Long short-term memory recurrent neural networks
    including our own fast CUDA kernel
  • Multidimensional LSTM (GPU only, there is no CPU version)
  • Memory management for large data sets
  • Work distribution across multiple devices
  • Flexible and fast architecture which allows all kinds of encoder-attention-decoder models

See documentation <http://returnn.readthedocs.io/>.
See basic usage <https://returnn.readthedocs.io/en/latest/basic_usage.html>

and technological overview <https://returnn.readthedocs.io/en/latest/tech_overview.html>__.

Here is the video recording of a RETURNN overview talk <https://www-i6.informatik.rwth-aachen.de/web/Software/returnn/downloads/workshop-2019-01-29/01.recording.cut.mp4>_
(slides <https://www-i6.informatik.rwth-aachen.de/web/Software/returnn/downloads/workshop-2019-01-29/01.returnn-overview.session1.handout.v1.pdf>,
exercise sheet <https://www-i6.informatik.rwth-aachen.de/web/Software/returnn/downloads/workshop-2019-01-29/01.exercise_sheet.pdf>
;
hosted by eBay).

There are many example demos <https://github.com/rwth-i6/returnn/blob/master/demos/>_
which work on artificially generated data,
i.e. they should work as-is.

There are some real-world examples <https://github.com/rwth-i6/returnn-experiments>_
such as setups for speech recognition on the Switchboard or LibriSpeech corpus.

Some benchmark setups against other frameworks
can be found here <https://github.com/rwth-i6/returnn-benchmarks>.
The results are in the RETURNN paper 2016 <https://arxiv.org/abs/1608.00895>
.
Performance benchmarks of our LSTM kernel vs CuDNN and other TensorFlow kernels
are in TensorFlow LSTM benchmark <https://returnn.readthedocs.io/en/latest/tf_lstm_benchmark.html>__.

There is also a wiki <https://github.com/rwth-i6/returnn/wiki>.
Questions can also be asked on
StackOverflow using the RETURNN tag <https://stackoverflow.com/questions/tagged/returnn>
.

.. image:: https://travis-ci.org/rwth-i6/returnn.svg?branch=master
:target: https://travis-ci.org/rwth-i6/returnn

主要指标

概览
名称与所有者rwth-i6/returnn
主编程语言Python
编程语言Python (语言数: 4)
平台
许可证Other
所有者活动
创建于2016-06-14 14:37:14
推送于2025-06-04 15:01:36
最后一次提交2025-06-03 23:21:35
发布数3
最新版本名称v2.1.0-beta (发布于 2018-11-20 11:52:20)
第一版名称v2.0.1-beta (发布于 )
用户参与
星数367
关注者数26
派生数132
提交数12.1k
已启用问题?
问题数693
打开的问题数181
拉请求数909
打开的拉请求数27
关闭的拉请求数90
项目设置
已启用Wiki?
已存档?
是复刻?
已锁定?
是镜像?
是私有?