dungtd2403's picture
Upload A2C CartPole-v1 trained agent
de1383e
raw
history blame
6.92 kB
{
"policy_class": {
":type:": "<class 'abc.ABCMeta'>",
":serialized:": "gAWVOwAAAAAAAACMIXN0YWJsZV9iYXNlbGluZXMzLmNvbW1vbi5wb2xpY2llc5SMEUFjdG9yQ3JpdGljUG9saWN5lJOULg==",
"__module__": "stable_baselines3.common.policies",
"__doc__": "\n Policy class for actor-critic algorithms (has both policy and value prediction).\n Used by A2C, PPO and the likes.\n\n :param observation_space: Observation space\n :param action_space: Action space\n :param lr_schedule: Learning rate schedule (could be constant)\n :param net_arch: The specification of the policy and value networks.\n :param activation_fn: Activation function\n :param ortho_init: Whether to use or not orthogonal initialization\n :param use_sde: Whether to use State Dependent Exploration or not\n :param log_std_init: Initial value for the log standard deviation\n :param full_std: Whether to use (n_features x n_actions) parameters\n for the std instead of only (n_features,) when using gSDE\n :param use_expln: Use ``expln()`` function instead of ``exp()`` to ensure\n a positive standard deviation (cf paper). It allows to keep variance\n above zero and prevent it from growing too fast. In practice, ``exp()`` is usually enough.\n :param squash_output: Whether to squash the output using a tanh function,\n this allows to ensure boundaries when using gSDE.\n :param features_extractor_class: Features extractor to use.\n :param features_extractor_kwargs: Keyword arguments\n to pass to the features extractor.\n :param share_features_extractor: If True, the features extractor is shared between the policy and value networks.\n :param normalize_images: Whether to normalize images or not,\n dividing by 255.0 (True by default)\n :param optimizer_class: The optimizer to use,\n ``th.optim.Adam`` by default\n :param optimizer_kwargs: Additional keyword arguments,\n excluding the learning rate, to pass to the optimizer\n ",
"__init__": "<function ActorCriticPolicy.__init__ at 0x7f2a3d8aa9d0>",
"_get_constructor_parameters": "<function ActorCriticPolicy._get_constructor_parameters at 0x7f2a3d8aaa60>",
"reset_noise": "<function ActorCriticPolicy.reset_noise at 0x7f2a3d8aaaf0>",
"_build_mlp_extractor": "<function ActorCriticPolicy._build_mlp_extractor at 0x7f2a3d8aab80>",
"_build": "<function ActorCriticPolicy._build at 0x7f2a3d8aac10>",
"forward": "<function ActorCriticPolicy.forward at 0x7f2a3d8aaca0>",
"extract_features": "<function ActorCriticPolicy.extract_features at 0x7f2a3d8aad30>",
"_get_action_dist_from_latent": "<function ActorCriticPolicy._get_action_dist_from_latent at 0x7f2a3d8aadc0>",
"_predict": "<function ActorCriticPolicy._predict at 0x7f2a3d8aae50>",
"evaluate_actions": "<function ActorCriticPolicy.evaluate_actions at 0x7f2a3d8aaee0>",
"get_distribution": "<function ActorCriticPolicy.get_distribution at 0x7f2a3d8aaf70>",
"predict_values": "<function ActorCriticPolicy.predict_values at 0x7f2a3d8af040>",
"__abstractmethods__": "frozenset()",
"_abc_impl": "<_abc._abc_data object at 0x7f2a3d8ac9c0>"
},
"verbose": 0,
"policy_kwargs": {
":type:": "<class 'dict'>",
":serialized:": "gAWVgQAAAAAAAAB9lCiMD29wdGltaXplcl9jbGFzc5SME3RvcmNoLm9wdGltLnJtc3Byb3CUjAdSTVNwcm9wlJOUjBBvcHRpbWl6ZXJfa3dhcmdzlH2UKIwFYWxwaGGURz/vrhR64UeujANlcHOURz7k+LWI42jxjAx3ZWlnaHRfZGVjYXmUSwB1dS4=",
"optimizer_class": "<class 'torch.optim.rmsprop.RMSprop'>",
"optimizer_kwargs": {
"alpha": 0.99,
"eps": 1e-05,
"weight_decay": 0
}
},
"observation_space": {
":type:": "<class 'gym.spaces.box.Box'>",
":serialized:": "gAWVdwEAAAAAAACMDmd5bS5zcGFjZXMuYm94lIwDQm94lJOUKYGUfZQojAVkdHlwZZSMBW51bXB5lGgFk5SMAmY0lImIh5RSlChLA4wBPJROTk5K/////0r/////SwB0lGKMBl9zaGFwZZRLBIWUjANsb3eUjBJudW1weS5jb3JlLm51bWVyaWOUjAtfZnJvbWJ1ZmZlcpSTlCiWEAAAAAAAAACamZnA//9//1B31r7//3//lGgKSwSFlIwBQ5R0lFKUjARoaWdolGgSKJYQAAAAAAAAAJqZmUD//39/UHfWPv//f3+UaApLBIWUaBV0lFKUjA1ib3VuZGVkX2JlbG93lGgSKJYEAAAAAAAAAAEBAQGUaAeMAmIxlImIh5RSlChLA4wBfJROTk5K/////0r/////SwB0lGJLBIWUaBV0lFKUjA1ib3VuZGVkX2Fib3ZllGgSKJYEAAAAAAAAAAEBAQGUaCFLBIWUaBV0lFKUjApfbnBfcmFuZG9tlE51Yi4=",
"dtype": "float32",
"_shape": [
4
],
"low": "[-4.8000002e+00 -3.4028235e+38 -4.1887903e-01 -3.4028235e+38]",
"high": "[4.8000002e+00 3.4028235e+38 4.1887903e-01 3.4028235e+38]",
"bounded_below": "[ True True True True]",
"bounded_above": "[ True True True True]",
"_np_random": null
},
"action_space": {
":type:": "<class 'gym.spaces.discrete.Discrete'>",
":serialized:": "gAWVggAAAAAAAACME2d5bS5zcGFjZXMuZGlzY3JldGWUjAhEaXNjcmV0ZZSTlCmBlH2UKIwBbpRLAowGX3NoYXBllCmMBWR0eXBllIwFbnVtcHmUaAeTlIwCaTiUiYiHlFKUKEsDjAE8lE5OTkr/////Sv////9LAHSUYowKX25wX3JhbmRvbZROdWIu",
"n": 2,
"_shape": [],
"dtype": "int64",
"_np_random": null
},
"n_envs": 1,
"num_timesteps": 0,
"_total_timesteps": 0,
"_num_timesteps_at_start": 0,
"seed": null,
"action_noise": null,
"start_time": null,
"learning_rate": 0.001,
"tensorboard_log": null,
"lr_schedule": {
":type:": "<class 'function'>",
":serialized:": "gAWV6QIAAAAAAACMF2Nsb3VkcGlja2xlLmNsb3VkcGlja2xllIwOX21ha2VfZnVuY3Rpb26Uk5QoaACMDV9idWlsdGluX3R5cGWUk5SMCENvZGVUeXBllIWUUpQoSwFLAEsASwFLAUsTQwSIAFMAlE6FlCmMAV+UhZSMWy9ob21lL2R1bmcvLmNvbmRhL2VudnMvcmxfem9vL2xpYi9weXRob24zLjkvc2l0ZS1wYWNrYWdlcy9zdGFibGVfYmFzZWxpbmVzMy9jb21tb24vdXRpbHMucHmUjARmdW5jlEuCQwIAAZSMA3ZhbJSFlCl0lFKUfZQojAtfX3BhY2thZ2VfX5SMGHN0YWJsZV9iYXNlbGluZXMzLmNvbW1vbpSMCF9fbmFtZV9flIwec3RhYmxlX2Jhc2VsaW5lczMuY29tbW9uLnV0aWxzlIwIX19maWxlX1+UjFsvaG9tZS9kdW5nLy5jb25kYS9lbnZzL3JsX3pvby9saWIvcHl0aG9uMy45L3NpdGUtcGFja2FnZXMvc3RhYmxlX2Jhc2VsaW5lczMvY29tbW9uL3V0aWxzLnB5lHVOTmgAjBBfbWFrZV9lbXB0eV9jZWxslJOUKVKUhZR0lFKUjBxjbG91ZHBpY2tsZS5jbG91ZHBpY2tsZV9mYXN0lIwSX2Z1bmN0aW9uX3NldHN0YXRllJOUaB99lH2UKGgWaA2MDF9fcXVhbG5hbWVfX5SMGWNvbnN0YW50X2ZuLjxsb2NhbHM+LmZ1bmOUjA9fX2Fubm90YXRpb25zX1+UfZSMDl9fa3dkZWZhdWx0c19flE6MDF9fZGVmYXVsdHNfX5ROjApfX21vZHVsZV9flGgXjAdfX2RvY19flE6MC19fY2xvc3VyZV9flGgAjApfbWFrZV9jZWxslJOURz9QYk3S8an8hZRSlIWUjBdfY2xvdWRwaWNrbGVfc3VibW9kdWxlc5RdlIwLX19nbG9iYWxzX1+UfZR1hpSGUjAu"
},
"_last_obs": null,
"_last_episode_starts": null,
"_last_original_obs": null,
"_episode_num": 0,
"use_sde": false,
"sde_sample_freq": -1,
"_current_progress_remaining": 1,
"ep_info_buffer": null,
"ep_success_buffer": null,
"_n_updates": 0,
"n_steps": 10000,
"gamma": 0.99,
"gae_lambda": 1.0,
"ent_coef": 0.0,
"vf_coef": 0.5,
"max_grad_norm": 1.2,
"normalize_advantage": false
}