/
envoy.istio
/
mindspore
Обзор
Документация
Войти
/
envoy.istio
/
mindspore
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
docs/api/api_python/train/mindspore.train.FlopsUtilizationCollector.rst
34 строки
2 KB
zhouyaqiang2
add flops collection
15 июн 2024, 11:50
15 июн 2024, 11:50
0968bdd
Код
Авторство
О чём код?
mindspore.train.FlopsUtilizationCollector ========================================= .. py:class:: mindspore.train.FlopsUtilizationCollector(data_size, computility=1, full_flops=True) FlopsUtilizationCollector接口统计模型利用率信息MFU，硬件利用率信息HFU。 当前接口只统计MatMul、BatchMatMul、FlashAttentionScore、Conv2D算子的正反向flops信息。 只支持静态图静态shape模式。 参数： - **data_size** (int) - 表示每隔多少个step打印一次信息。 - **computility** (int) - 表示每张计算卡的峰值算力。默认值： ``1`` 。 - **full_flops** (bool) - 表示是否统计完整的模型信息，如果设置为True，会统计完整的模型信息，如果设置为False，将会统计对应每张卡的分片模型信息。默认值： ``True`` 。 异常： - **TypeError** - `data_size` 不是正整数。 - **TypeError** - `full_flops` 不是布尔类型。 - **AssertionError** - 不是静态图或者不是静态shape。 .. py:method:: epoch_begin(run_context) 在epoch开始时记录时间。 参数： - **run_context** (RunContext) - 包含模型的相关信息。详情请参考 :class:`mindspore.train.RunContext`。 .. py:method:: epoch_end(run_context) 在epoch结束时打印模型利用率信息MFU，硬件利用率信息HFU。 参数： - **run_context** (RunContext) - 包含模型的相关信息。详情请参考 :class:`mindspore.train.RunContext`。