你好:
CNMIX是构建与框架(Cambricon TensorFlow和Cambricon PyTorch)上的混合精度训练策略库,能够将原框架模型转换成自适应量化的方式进行训练。
pyamp
算子的输入输出,根据AMP内部使用策略使用FP16或FP32计算。
算子计算时,不同的算子有自己的autocast计算规则。
请登录后评论