MCPcopy Create free account
hub / github.com/pytorch/examples / setup

Function setup

distributed/FSDP/utils/train_utils.py:11–13  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

9g_gigabyte = 1024**3
10
11def setup():
12 # initialize the process group
13 dist.init_process_group("nccl")
14
15
16def cleanup():

Callers 1

fsdp_mainFunction · 0.90

Calls

no outgoing calls

Tested by

no test coverage detected