Exemplo n.º 1
0
 def test_d3pg_train(self):
     CONFIG_PATH = 'tests/bipedal/config.yml'
     config = read_config(CONFIG_PATH)
     config['model'] = 'd3pg'
     train(config=config)
 def test_d4pg_train(self):
     CONFIG_PATH = 'tests/lunar_lander_continous/config.yml'
     config = read_config(CONFIG_PATH)
     config['model'] = 'd4pg'
     train(config)
Exemplo n.º 3
0
 def test_d3pg_train_prioritized(self):
     CONFIG_PATH = 'tests/pendulum/config.yml'
     config = read_config(CONFIG_PATH)
     config['model'] = 'd3pg'
     config['replay_memory_prioritized'] = 1
     train(config)
 def test_d4pg_train_prioritized(self):
     CONFIG_PATH = 'tests/lunar_lander_continous/config.yml'
     config = read_config(CONFIG_PATH)
     config['model'] = 'd4pg'
     config['replay_memory_prioritized'] = 1
     train(config)
Exemplo n.º 5
0
 def test_d4pg_train(self):
     CONFIG_PATH = "tests/pendulum/config.yml"
     config = read_config(CONFIG_PATH)
     config['model'] = 'd4pg'
     train(config=config)