-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.py
100 lines (77 loc) · 2.46 KB
/
config.py
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
'''
This file contains the configs used for Model creation and training. You need to give your best hyperparameters and the configs you used to get the best results for
every environment and experiment. These configs will be automatically loaded and used to create and train your model in our servers.
'''
#You can add extra keys or modify to the values of the existing keys in bottom level of the dictionary.
#DONT CHANGE THE OVERALL STRUCTURE OF THE DICTIONARY.
configs = {
'Ant-v4': {
"imitation":{
#You can add or change the keys here
"hyperparameters": {
"ntraj" : 50,
"maxtraj" : 400
},
"num_iteration": 10,
},
"RL":{
#An example set of hyperparameters is given below
#You can add or change the hyperparameters and other keys here here
"hyperparameters": {
'n_layers': 4,
'hidden_size': 64,
'learning_rate': 0.001,
'batch_size': 3000,
'max_ep_len': None,
'discount' : 0.9,
},
"num_iteration": 200,
},
"imitation-RL":{
#You can add or change the keys here
"hyperparameters": {
},
"num_iteration": 100,
}
},
'Hopper-v4': {
"imitation":{
#You can add or change the keys here
"hyperparameters": {
"ntraj" : 50,
"maxtraj" : 400
},
"num_iteration": 75,
},
"RL":{
#You can add or change the keys here
"hyperparameters": {
"ntraj" : 50,
"maxtraj" : 400
},
"num_iteration": 100,
},
},
'HalfCheetah-v4': {
"imitation":{
#You can add or change the keys here
"hyperparameters": {
"ntraj" : 50,
"maxtraj" : 400
},
"num_iteration": 20,
},
"RL":{
#You can add or change the keys here
"hyperparameters": {
},
"num_iteration": 100,
},
"imitation-RL":{
#You can add or change the keys here
"hyperparameters": {
},
"num_iteration": 100,
}
},
}