transformer_engine

将模型优化器量化模型转换为transformer_engine。

函数

convert_to_transformer_engine

Model Optimizer量化的模型转换为transformers_engine

convert_to_transformer_engine(model)

Model Optimizer量化的模型转换为transformers_engine

Parameters:

模型 (模块) –