ksuit.utils.training.seed

Attributes

logger

Functions

set_seed(seed)

Set the seed for random number generation for Python random, numpy,

Module Contents

ksuit.utils.training.seed.logger

ksuit.utils.training.seed.set_seed(seed)

Set the seed for random number generation for Python random, numpy, torch and torch.cuda, if available.

Parameters:

seed (int) – Seed value.

Return type:

None