def policy_fn(name, ob_space, ac_space):
if args.mode == "SENSOR":
return mlp_policy.MlpPolicy(name=name, ob_space=ob_space, ac_space=ac_space, hid_size=64, num_hid_layers=2)
else:
#return fuse_policy.FusePolicy(name=name, ob_space=ob_space, sensor_space=sensor_space, ac_space=ac_space, save_per_acts=10000, session=sess)
#else:
return cnn_policy.CnnPolicy(name=name, ob_space=ob_space, ac_space=ac_space, save_per_acts=10000, session=sess, kind='small')
def policy_fn(name, ob_space, ac_space):
#return mlp_policy.MlpPolicy(name=name, ob_space=sensor_space, ac_space=ac_space, hid_size=64, num_hid_layers=2)
return cnn_policy.CnnPolicy(name=name,
ob_space=ob_space,
ac_space=ac_space,
save_per_acts=10000,
session=sess,
kind='small')
def policy_fn(name, ob_space, ac_space):
if args.mode == "SENSOR":
return mlp_policy.MlpPolicy(name=name,
ob_space=ob_space,
ac_space=ac_space,
hid_size=64,
num_hid_layers=2)
else:
return cnn_policy.CnnPolicy(name=name,
ob_space=ob_space,
ac_space=ac_space,
session=sess,
kind='small')
def policy_fn(name, ob_space, ac_space):
return cnn_policy.CnnPolicy(name=name, ob_space=ob_space, ac_space=ac_space, session=sess, kind='small')