Random State in TensorFlow

1
2
3
4
5
6
7
8
9
10
11
import os
import tensorflow as tf
from tfdeterminism import patch

RANDOM_STATE = 42

patch()
os.environ['PYTHONHASHSEED"]         = str(RANDOM_STATE)
os.environ['TF_DETERMINISTIC_OPS']   = '1'
os.environ['TF_CUDNN_DETERMINISTIC'] = '1'
tf.keras.utils.set_random_seed(RANDOM_STATE)

1
2
3
4
5
6
7
import random
import numpy as np
import tensorflow as tf

random.seed(seed)
np.random.seed(seed)
tf.random.set_seed(seed)