使用 PyTorch 进行数据集量化

Question

我目前正在使用一种工具将 CNN 模型部署到 FPGA 中，该工具仅支持少数位量化模型（使用量化感知训练方法）。

因此，例如，我需要将我的数据集量化为 4 位整数（来自 fp32 值）。我的虚拟数据集由正弦/余弦运算结果数据组成。

有没有办法轻松做到？或者一个教程来做到这一点？