Python PyTorch FileStore用法及代碼示例

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

用法:

class torch.distributed.FileStore

參數：

• file_name(str) -存儲鍵值對的文件的路徑

• world_size(int,可選的) -使用存儲的進程總數。默認值為 -1(負值表示商店用戶數量不固定)。

使用文件存儲底層鍵值對的存儲實現。

例子：

>>> import torch.distributed as dist
>>> store1 = dist.FileStore("/tmp/filestore", 2)
>>> store2 = dist.FileStore("/tmp/filestore", 2)
>>> # Use any of the store methods from either the client or server after initialization
>>> store1.set("first_key", "first_value")
>>> store2.get("first_key")