config.json

{
  "env_name": "SpaceInvadersNoFrameskip-v4",
  "max_episode_steps": 10000,
  "latent_dim": 32,
  "hidden_dim": 256,
  "num_mixtures": 5,
  "action_dim": 6,
  "vae_epochs": 10,
  "vae_batch_size": 32,
  "vae_lr": 0.0001,
  "num_random_episodes": 100,
  "rnn_epochs": 20,
  "rnn_batch_size": 16,
  "rnn_lr": 0.0001,
  "sequence_length": 64,
  "population_size": 64,
  "num_generations": 10,
  "sigma": 0.5,
  "eval_episodes": 10,
  "record_video_every": 10,
  "device": "cuda"
}