File size: 2,593 Bytes
a1c2629 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 |
exp_config = {
'env': {
'manager': {
'episode_num': float("inf"),
'max_retry': 1,
'retry_type': 'reset',
'auto_reset': True,
'step_timeout': None,
'reset_timeout': None,
'retry_waiting_time': 0.1,
'cfg_type': 'BaseEnvManagerDict'
},
'stop_value': 240,
'collector_env_num': 8,
'evaluator_env_num': 8,
'env_id': 'LunarLander-v2',
'n_evaluator_episode': 8
},
'policy': {
'model': {
'obs_shape': 8,
'action_shape': 4
},
'learn': {
'learner': {
'train_iterations': 1000000000,
'dataloader': {
'num_workers': 0
},
'log_policy': True,
'hook': {
'load_ckpt_before_run': '',
'log_show_after_iter': 100,
'save_ckpt_after_iter': 10000,
'save_ckpt_after_run': True
},
'cfg_type': 'BaseLearnerDict'
},
'update_per_collect': 1,
'batch_size': 160,
'learning_rate': 0.0003,
'betas': [0.9, 0.999],
'eps': 1e-08,
'grad_norm': 0.5,
'value_weight': 0.5,
'entropy_weight': 0.001,
'adv_norm': True,
'ignore_done': False
},
'collect': {
'collector': {},
'unroll_len': 1,
'discount_factor': 0.99,
'gae_lambda': 0.95,
'n_sample': 320
},
'eval': {
'evaluator': {
'eval_freq': 1000,
'render': {
'render_freq': -1,
'mode': 'train_iter'
},
'cfg_type': 'InteractionSerialEvaluatorDict',
'n_episode': 8,
'stop_value': 240
}
},
'other': {
'replay_buffer': {}
},
'on_policy': True,
'cuda': True,
'multi_gpu': False,
'bp_update_sync': True,
'traj_len_inf': False,
'type': 'a2c',
'priority': False,
'priority_IS_weight': False,
'cfg_type': 'A2CPolicyDict'
},
'exp_name': 'Lunarlander-v2-A2C',
'wandb_logger': {
'gradient_logger': True,
'video_logger': True,
'plot_logger': True,
'action_logger': True,
'return_logger': False
},
'seed': 0
}
|