MCPcopy Create free account
hub / github.com/AIRMEC/HECTOR / seed_worker

Function seed_worker

train.py:27–30  ·  view source on GitHub ↗
(worker_id)

Source from the content-addressed store, hash-verified

25 torch.backends.cudnn.deterministic = True
26
27def seed_worker(worker_id):
28 worker_seed = torch.initial_seed() % 2**32
29 np.random.seed(worker_seed)
30 random.seed(worker_seed)
31
32def evaluate_model(epoch, model, model_mol, device, loader, n_bins, writer, loss_fn, bins_values, train_BS, test_BS):
33 model.eval()

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected