14
14
# limitations under the License.
15
15
16
16
import argparse
17
+ import os
17
18
18
19
# EAGERx imports
19
20
from eagerx import Object , Bridge , Node , initialize , log
@@ -34,14 +35,15 @@ def example_classifier(name, eps, eval_eps, device, render=False):
34
35
"GymObject" ,
35
36
"pendulum" ,
36
37
sensors = ["image" , "observation" , "reward" , "done" ],
37
- gym_env_id = "Pendulum-v0" ,
38
- gym_rate = 20 ,
39
- gym_always_render = True ,
38
+ env_id = "Pendulum-v0" ,
39
+ rate = 20 ,
40
+ always_render = True ,
40
41
render_shape = [28 , 28 ],
41
42
)
42
43
43
44
# Define PID controller & classifier
44
- classifier = Node .make ("Classifier" , "classifier" , rate = 20 , cam_rate = 20 , data = "../data/with_actions.h5" )
45
+ dir_path = os .path .dirname (os .path .realpath (__file__ ))
46
+ classifier = Node .make ("Classifier" , "classifier" , rate = 20 , cam_rate = 20 , data = dir_path + "/../data/with_actions.h5" )
45
47
pid = Node .make ("PidController" , "pid" , rate = 20 , gains = [8 , 1 , 0 ], y_range = [- 4 , 4 ])
46
48
47
49
# Define graph (agnostic) & connect nodes
@@ -63,9 +65,9 @@ def example_classifier(name, eps, eval_eps, device, render=False):
63
65
bridge = Bridge .make ("GymBridge" , rate = 20 )
64
66
65
67
# Initialize Environment (agnostic graph + bridge)
66
- env = eagerx_gym .EagerGym (name = name , rate = 20 , graph = graph , bridge = bridge )
68
+ env = eagerx_gym .EagerxGym (name = name , rate = 20 , graph = graph , bridge = bridge )
67
69
if render :
68
- env .render (mode = ' human' )
70
+ env .render (mode = " human" )
69
71
70
72
# Initialize and train stable-baselines model
71
73
model = sb .SAC ("MlpPolicy" , env , verbose = 1 , device = device )
@@ -87,7 +89,7 @@ def example_classifier(name, eps, eval_eps, device, render=False):
87
89
parser .add_argument ("--name" , help = "Name of the environment" , type = str , default = "example" )
88
90
parser .add_argument ("--eps" , help = "Number of training episodes" , type = int , default = 200 )
89
91
parser .add_argument ("--eval_eps" , help = "Number of evaluation episodes" , type = int , default = 20 )
90
- parser .add_argument ("--render" , help = "Toggle rendering" , action = ' store_true' )
92
+ parser .add_argument ("--render" , help = "Toggle rendering" , action = " store_true" )
91
93
92
94
args = parser .parse_args ()
93
95
0 commit comments