Python make_tf_callable Beispiele

Beispiel #1

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

Beispiel #2

Datei: centralized_critic.py Projekt: 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

Beispiel #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

Beispiel #4

Datei: Centralized.py Projekt: devanderhoff/bluesky

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