tensorflow/keras 设定随机种子seed

需要tf2.0以上

还需要安装:pip install tensorflow-determinism

random.seed(seed)
np.random.seed(seed)
tf.random.set_seed(seed)
os.environ['PYTHONHASHSEED'] = str(seed)
tf.config.experimental.enable_op_determinism()

posted @ 2022-04-18 18:00  乌蝇哥  阅读(802)  评论(0编辑  收藏  举报