MCPcopy
hub / github.com/hpcaitech/ColossalAI / seed_worker

Method seed_worker

colossalai/booster/plugin/dp_plugin_base.py:61–65  ·  view source on GitHub ↗
(worker_id)

Source from the content-addressed store, hash-verified

59
60 # Deterministic dataloader
61 def seed_worker(worker_id):
62 worker_seed = seed
63 np.random.seed(worker_seed)
64 torch.manual_seed(worker_seed)
65 random.seed(worker_seed)
66
67 return DataLoader(
68 dataset,

Callers

nothing calls this directly

Calls 2

seedMethod · 0.45
manual_seedMethod · 0.45

Tested by

no test coverage detected