delu.random.set_state#

delu.random.set_state(state)[source]#

Set global random states in random, numpy and torch.

See get_state for a usage example.

Parameters:

state (Dict[str, Any]) – the global RNG states. Must be produced by get_state. The size of the list state['torch.cuda'] must be equal to the number of available cuda devices.

Return type:

None

Raises:

AssertionError – if torch.cuda.device_count() != len(state['torch.cuda'])

Parameters:

state (Dict[str, Any]) –

Return type:

None