Python PyTorch prof用法及代碼示例

本文簡要介紹python語言中 torch.distributed.elastic.metrics.prof 的用法。

用法:

torch.distributed.elastic.metrics.prof(fn=None, group='torchelastic')

@profile 裝飾器為它裝飾的函數發布持續時間.ms、計數、成功、失敗指標。指標名稱默認為函數的限定名稱 (class_name.def_name)。如果函數不屬於某個類，則使用葉模塊名稱代替。

用法

@metrics.prof
def x():
    pass

@metrics.prof(group="agent")
def y():
    pass