4.2.3.3. 设置不同的 BPU 架构

在定义网络之前设置 march 变量,例如:

from horizon_plugin_pytorch.march import March, set_march
set_march(March.BAYES)
qat_net = ...

可供设置的 march 有 March.BAYES 和 March.BERNOULLI2.