Python load_engine Examples

Example #1

File: test_bipedal.py Project: schatty/d4pg-pytorch

    def test_ddpg(self):
        CONFIG_PATH = 'tests/config_test.yml'
        config = read_config(CONFIG_PATH)

        config['env'] = 'BipedalWalker-v3'
        config['model'] = 'ddpg'
        config['state_dim'] = 24
        config['action_dim'] = 4

        engine = load_engine(config)
        engine.train()

Example #2

File: test_pendulum.py Project: zhan0903/d4pg-pytorch

    def test_ddpg(self):
        CONFIG_PATH = 'tests/config_test.yml'
        config = read_config(CONFIG_PATH)

        config['env'] = 'Pendulum-v0'
        config['model'] = 'ddpg'
        config['state_dim'] = 3
        config['action_dim'] = 1

        engine = load_engine(config)
        engine.train()

Example #3

File: test_lunar.py Project: zhan0903/d4pg-pytorch

    def test_ddpg(self):
        CONFIG_PATH = 'tests/config_test.yml'
        config = read_config(CONFIG_PATH)

        config['env'] = 'LunarLanderContinuous-v2'
        config['model'] = 'ddpg'
        config['state_dim'] = 8
        config['action_dim'] = 2

        engine = load_engine(config)
        engine.train()

Example #4

File: train.py Project: schatty/agents-systems

import argparse
from models.engine import load_engine
from utils.misc import read_config

parser = argparse.ArgumentParser(description='Run training')
parser.add_argument("--config", type=str, help="Path to the config file.")

if __name__ == "__main__":
    args = vars(parser.parse_args())
    config = read_config(args['config'])
    engine = load_engine(config)
    engine.train()