MCPcopy Create free account
hub / github.com/TEA-Lab/TwoByTwo / cleanup

Function cleanup

src/script/our_eval.py:40–41  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

38 torch.cuda.set_device(rank)
39
40def cleanup():
41 dist.destroy_process_group()
42
43
44def train(rank, world_size, conf):

Callers 1

trainFunction · 0.70

Calls

no outgoing calls

Tested by

no test coverage detected