def policy_fn(name, ob_space, ac_space):
return CnnPolicy(name=name, ob_space=env.observation_space, ac_space=env.action_space)
def policy_fn(name, ob_space, ac_space, sess=None, placeholders=None): # pylint: disable=W0613
return CnnPolicy(name=name,
ob_space=ob_space,
ac_space=ac_space,
sess=sess,
placeholders=placeholders)
def policy_fn(name, ob_space, ac_space): #pylint: disable=W0613
return CnnPolicy(name=name, ob_space=env.observation_space, ac_space=env.action_space)
def policy_fn(name, ob_space, ac_space): #pylint: disable=W0613
return CnnPolicy(name=name, ob_space=box, ac_space=discrete)