File size: 3,171 Bytes
f77ce07
 
 
 
 
 
 
 
 
 
 
 
 
61f40f3
f77ce07
 
 
61f40f3
f77ce07
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
61f40f3
f77ce07
61f40f3
 
f77ce07
 
 
 
 
 
 
 
61f40f3
f77ce07
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
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': 12000,
        'n_evaluator_episode': 8,
        'env_id': 'HalfCheetah-v3',
        'collector_env_num': 1,
        'evaluator_env_num': 8,
        'env_wrapper': 'mujoco_default'
    },
    'policy': {
        'model': {
            'twin_critic': True,
            'action_space': 'reparameterization',
            'obs_shape': 17,
            'action_shape': 6,
            'actor_head_hidden_size': 256,
            'critic_head_hidden_size': 256
        },
        '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': 256,
            'learning_rate_q': 0.001,
            'learning_rate_policy': 0.001,
            'learning_rate_alpha': 0.0003,
            'target_theta': 0.005,
            'discount_factor': 0.99,
            'alpha': 0.2,
            'auto_alpha': False,
            'log_space': True,
            'target_entropy': None,
            'ignore_done': True,
            'init_w': 0.003,
            'reparameterization': True
        },
        'collect': {
            'collector': {},
            'n_sample': 1,
            'unroll_len': 1,
            'collector_logit': False
        },
        'eval': {
            'evaluator': {
                'eval_freq': 1000,
                'render': {
                    'render_freq': -1,
                    'mode': 'train_iter'
                },
                'figure_path': None,
                'cfg_type': 'InteractionSerialEvaluatorDict',
                'stop_value': 12000,
                'n_episode': 8
            }
        },
        'other': {
            'replay_buffer': {
                'replay_buffer_size': 1000000
            }
        },
        'on_policy': False,
        'cuda': True,
        'multi_gpu': False,
        'bp_update_sync': True,
        'traj_len_inf': False,
        'type': 'sac',
        'priority': False,
        'priority_IS_weight': False,
        'random_collect_size': 10000,
        'transition_with_policy_data': True,
        'multi_agent': False,
        'cfg_type': 'SACPolicyDict',
        'command': {}
    },
    'exp_name': 'HalfCheetah-v3-SAC',
    'seed': 0,
    'wandb_logger': {
        'gradient_logger': True,
        'video_logger': True,
        'plot_logger': True,
        'action_logger': True,
        'return_logger': False
    }
}