文件名称:TQT:TQT的pytorch实施
文件大小:252KB
文件格式:ZIP
更新时间:2024-04-22 04:00:15
Python
TQT TQT的pytorch实施。 注意 现在在可用availabel! 通过此程序包量化的网络可以在找到。 TQT的模块 TQT功能 function是torch.nn.modules的重新torch.nn.modules 。 除了原始函数中使用的所有args外,量化函数还获得2种可选参数: bit_width和retrain 。 bit_width有2种类型:权重/偏差或激活。 如果retrain为True ,则模块将处于retrain模式,且log2_t可训练。 否则,在静态模式下, log2_t由初始化确定并且不可训练。 TQT阈值 提供3种初始化阈值的方法: init_max , init_kl_j , init_3sd 。 要正确初始化权重和阈值,请遵循使用TQT建立网络的方法。 xxxxx.xxx 。 众所周知,输入(k1-b-m1-p)乘以权重(k2-b-m2
【文件预览】:
TQT-main
----example()
--------lenet_via_wrapper.py(1KB)
--------lenet_n.py(959B)
--------lenet_q.py(1KB)
--------resnet20.py(4KB)
----docs()
--------source()
----tqt()
--------utils.py(1KB)
--------function()
--------__init__.py(109B)
--------threshold()
--------wrapper.py(5KB)
--------config.py(2KB)
--------__version__.py(239B)
--------.vscode()
----style.md(2KB)
----LICENSE(1KB)
----makefile(85B)
----setup.py(1KB)
----.gitignore(2KB)
----README.md(7KB)
----save_qnet.pth(248KB)
----.vscode()
--------settings.json(61B)