mirror of
https://github.com/vale981/ray
synced 2025-03-06 18:41:40 -05:00
15 lines
345 B
Python
15 lines
345 B
Python
from __future__ import absolute_import
|
|
from __future__ import division
|
|
from __future__ import print_function
|
|
|
|
import numpy as np
|
|
import random
|
|
from ray.rllib.utils import try_import_tf
|
|
|
|
tf = try_import_tf()
|
|
|
|
|
|
def seed(np_seed=0, random_seed=0, tf_seed=0):
|
|
np.random.seed(np_seed)
|
|
random.seed(random_seed)
|
|
tf.set_random_seed(tf_seed)
|