tf2.3_rep

tf2.3系、ubuntu環境下での再現性の保証

下記の関数を定義し、モデルの定義まえに呼び出す。

def reproduct(s):
    os.environ['PYTHONHASHSEED'] = '0'
    os.environ['TF_DETERMINISTIC_OPS'] = '1'
    os.environ['TF_CUDNN_DETERMINISTIC'] = '1'
    tensorflow.random.set_seed(s)
    np.random.seed(s)
    random.seed(s)
    tensorflow.config.threading.set_inter_op_parallelism_threads(1)
    tensorflow.config.threading.set_intra_op_parallelism_threads(1)