def common_layers():
cur_layers = []
if FLAGS.flatten_non_batch_time_dims:
cur_layers = [
layers.Div(divisor=255.0),
layers.Flatten(num_axis_to_keep=2)
]
body = [layers.Dense(64), layers.Tanh(), layers.Dense(64), layers.Tanh()]
return cur_layers + body
def common_layers():
# TODO(afrozm): Refactor.
if "NoFrameskip" in FLAGS.env_problem_name:
return atari_layers()
cur_layers = []
if FLAGS.flatten_dims:
cur_layers = [
layers.Div(divisor=255.0),
layers.Flatten(num_axis_to_keep=2)
]
body = [layers.Dense(64), layers.Tanh(), layers.Dense(64), layers.Tanh()]
return cur_layers + body
def common_layers():
# TODO(afrozm): Refactor.
if "NoFrameskip" in FLAGS.env_problem_name:
return atari_layers()
return [layers.Dense(64), layers.Tanh(), layers.Dense(64), layers.Tanh()]
