imitation/benchmarking/dagger_seals_half_cheetah_best_hp_eval.json at ce8c87ddace0017801e6a5e8fcfcd2ca0dc24cf7 · HumanCompatibleAI/imitation · GitHub

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
{
  "bc": {
    "batch_size": 16,
    "l2_weight": 0.0001,
    "optimizer_cls": {
      "py/type": "torch.optim.adam.Adam"
    },
    "optimizer_kwargs": {
      "lr": 0.001
    },
    "train_kwargs": {
      "log_interval": 500,
      "n_batches": null,
      "n_epochs": 5
    }
  },
  "dagger": {
    "beta_schedule": {
      "decay_probability": 0.7,
      "py/object": "imitation.algorithms.dagger.ExponentialBetaSchedule"
    },
    "rollout_round_min_episodes": 5,
    "total_timesteps": 60000,
    "use_offline_rollouts": false
  },
  "demonstrations": {
    "source": "huggingface",
    "algo_name": "ppo",
    "n_expert_demos": null
  },
  "policy": {
    "policy_cls": {
      "py/type": "imitation.policies.base.FeedForward32Policy"
    },
    "policy_kwargs": {
      "features_extractor_class": {
        "py/type": "imitation.policies.base.NormalizeFeaturesExtractor"
      },
      "features_extractor_kwargs": {
        "normalize_class": {
          "py/type": "imitation.util.networks.RunningNorm"
        }
      }
    }
  },
  "policy_evaluation": {
    "n_episodes_eval": 50
  },
  "environment": {
    "gym_id": "seals/HalfCheetah-v1"
  }
}