tensor_rnn:基于张量分解的NN和RNN参数的实现

时间:2024-06-04 09:12:14
【文件属性】:

文件名称: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)

网友评论