def make_model(): def make_ac(): return ActorCriticTwoMLP(**actor_critic_args) return AgentModel(make_ac, distribution_str, make_obs_encoder=DeepConvEncoder)
def make_model(): def make_ac(): return OneLayerActorCritic(**actor_critic_args) model = AgentModel(make_ac, distribution_str, make_obs_encoder=make_encoder) return model
def make_model(): def make_ac(): return ActorCriticTwoMLP(**actor_critic_args) model = AgentModel( make_ac, distribution_str, make_obs_normalizer=True, make_reward_normalizer=True, make_value_normalizer=True ) return model
def make_model(): def make_ac(): return OneLayerActorCritic(**actor_critic_args) model = AgentModel( make_ac, distribution_str, make_obs_normalizer=True, make_obs_encoder=make_encoder, make_reward_normalizer=True, make_value_normalizer=True ) return model
def make_model(): def make_ac(): return ActorCriticTwoMLP(**actor_critic_args) model = AgentModel(make_ac, distribution_str) return model