文件名称:tensor_rnn:基于张量分解的NN和RNN参数的实现
文件大小:22KB
文件格式:ZIP
更新时间:2024-06-04 09:12:14
machine-learning compression deep-learning pytorch rnn
张量RNN 基于张量分解的NN和RNN参数的实现 快速开始 安装python >= 3.0 安装pytorch >= 3.0 pip install -e . 或python setup.py install 运行示例脚本 转到示例文件夹 cd example/polymusic 转到数据文件夹,下载腌制的数据集并返回。 cd data && ./download_data.sh && cd .. 运行任何示例脚本 python run_ttgru.py 有关用法,请参见example/polymusic/poly_allrnn.py的代码 模组 线性层 TuckerLinear CPLinear TTLinear 双线性层 CPBilinear TuckerBilinear (TODO) RNN层 StatefulCPLSTMCell StatefulCPGRUCe
【文件预览】:
tensor_rnn-master
----tensor_rnn()
--------utils()
--------modules()
--------example()
----LICENSE(1KB)
----setup.py(490B)
----README.md(1KB)
----.gitignore(1KB)