Python PyTorch record用法及代碼示例

本文簡要介紹python語言中 torch.distributed.elastic.multiprocessing.errors.record 的用法。

用法:

torch.distributed.elastic.multiprocessing.errors.record(fn, error_handler=None)

使用提供的 error_handler 記錄修飾函數中發生的錯誤/異常的語法糖。

使用這個裝飾器相當於：

error_handler = get_error_handler()
error_handler.initialize()
try:
   foobar()
except ChildFailedError as e:
   _, failure = e.get_first_failure()
   error_handler.dump_error_file(failure.error_file, failure.exitcode)
   raise
except Exception as e:
   error_handler.record(e)
   raise

重要的

在頂層方法中每個進程使用一次此裝飾器，通常這是主要方法。

示例

@record
def main():
    pass

if __name__=="__main__":
   main()