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.
在定义网络之前设置 march 变量,例如:
from horizon_plugin_pytorch.march import March, set_march
set_march(March.BAYES)
qat_net = ...
可供设置的 march 有 March.BAYES 和 March.BERNOULLI2.