diff --git a/run_scripts/baselines.py b/run_scripts/baselines.py index 04f5ad7bcb2c22c2c0f7d5652ee4479012eb3118..9afa1c14b4c35505c1dec5f3582332c50a829f84 100644 --- a/run_scripts/baselines.py +++ b/run_scripts/baselines.py @@ -9,7 +9,7 @@ import time start = time.time() c_32 = ew.Config() -c_32.name = 'FirstLayerBase' +c_32.name = 'Deep' c_32.render = False c_32.force_cpu = True c_32.env = gym.make('CartPole-v0') @@ -36,6 +36,21 @@ c_512.net_layout = [512, 32] c_1024 = copy.deepcopy(c_32) c_1024.net_layout = [1024, 32] +cd_3 = copy.deepcopy(c_32) +cd_3.net_layout[128, 64, 32] + +cd_4 = copy.deepcopy(c_32) +cd_4.net_layout[128, 64, 32, 32] + +cd_128 = copy.deepcopy(c_32) +cd_128.net_layout[128, 128, 128] + +cd_256 = copy.deepcopy(c_32) +cd_256.net_layout[256, 256, 256] + +cd_512 = copy.deepcopy(c_32) +cd_512.net_layout[512, 512, 512] + conf = c_32 conf.conf_to_name()