test-data/simple/config.ron010066400017500001750000000025351426344715200144220ustar 00000000000000TrainConfig( version: 3, env: EnvConfig( kwargs: "{}", id: "BevyMultiSnake", validate: true, ), net: RogueNetConfig( embd_pdrop: 0.0, resid_pdrop: 0.0, attn_pdrop: 0.0, n_layer: 2, n_head: 2, d_model: 4, pooling: None, relpos_encoding: None, d_qk: 16, translation: None, ), optim: OptimizerConfig( lr: 0.001, bs: 1024, weight_decay: 0.0, micro_bs: None, anneal_lr: true, update_epochs: 3, max_grad_norm: 2.0, ), ppo: PPOConfig( gae: true, gamma: 0.99, gae_lambda: 0.95, norm_adv: true, clip_coef: 0.2, clip_vloss: true, ent_coef: 0.1, vf_coef: 0.5, target_kl: None, anneal_entropy: true, ), rollout: RolloutConfig( steps: 16, num_envs: 128, processes: 4, ), eval: None, vf_net: None, name: "config", seed: 1, total_timesteps: 25000, max_train_time: None, torch_deterministic: true, cuda: true, track: false, wandb_project_name: "autochess", wandb_entity: "cswinter", capture_samples: None, capture_logits: false, capture_samples_subsample: 1, trial: None, data_dir: ".", cuda_empty_cache: false, )test-data/simple/state.ron010066400017500001750000000016271426344715200142760ustar 00000000000000State( step: 24576, restart: 2, next_eval_step: None, agent: "", value_function: None, optimizer: "", vf_optimizer: None, obs_space: ObsSpace( global_features: [], entities: { "SnakeSegment": Entity( features: [ "x", "y", ], ), "Head": Entity( features: [ "x", "y", ], ), "Food": Entity( features: [ "x", "y", ], ), }, ), action_space: { "action": CategoricalActionSpace( index_to_label: [ "Up", "Down", "Left", "Right", ], ), }, )test-data/simple/state.agent.msgpack010066400017500001750000000155321426344715200162220ustar 000000000000009)embedding.embeddings.SnakeSegment.0.count __tensor__torchdtypew¥T>M,m>VS>(embedding.embeddings.SnakeSegment.1.bias __tensor__torchdtypeg@X=*embedding.embeddings.SnakeSegment.3.weight __tensor__torchdtypeG;kP;!embedding.embeddings.Head.0.count __tensor__torchdtype*S_S=;?Ml> embedding.embeddings.Head.1.bias __tensor__torchdtype(> -w>l$F>/? embedding.embeddings.Food.1.bias __tensor__torchdtype2*/"embedding.embeddings.Food.3.weight __tensor__torchdtypezm=e6<v7 backbone.blocks.0.attn.proj.bias __tensor__torchdtypeҮ'ʆ<;gަ7><p_<Ρ9ݼ&zI%