Exemplos de make_tf_callable em Python

Exemplo n.º 1

 def __init__(self):
     self.compute_central_vf = make_tf_callable(self.get_session())(
         self.model.central_value_function)

Exemplo n.º 2

Arquivo: centralized_critic.py Projeto: MetaMind/ray-internal

 def __init__(self):
     if self.config["framework"] != "torch":
         self.compute_central_vf = make_tf_callable(self.get_session())(
             self.model.central_value_function)
     else:
         self.compute_central_vf = self.model.central_value_function

Exemplo n.º 3

 def __init__(self):
     if not self.config["use_pytorch"]:
         self.compute_central_vf = make_tf_callable(self.get_session())(
             self.model.central_value_function)
     else:
         self.compute_central_vf = self.model.central_value_function

Exemplo n.º 4

Arquivo: Centralized.py Projeto: devanderhoff/bluesky

 def __init__(self):
     self.compute_central_vf = make_tf_callable(self.get_session(), dynamic_shape=True)(
         self.model.central_value_function)