Upload folder using huggingface_hub

.gitattributes

@@ -33,3 +33,8 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+a100_100x100.mp4 filter=lfs diff=lfs merge=lfs -text
+a20_40x40.mp4 filter=lfs diff=lfs merge=lfs -text
+a20_b4_40x40.mp4 filter=lfs diff=lfs merge=lfs -text
+a20_r4_40x40.mp4 filter=lfs diff=lfs merge=lfs -text
+replay.mp4 filter=lfs diff=lfs merge=lfs -text

.summary/0/events.out.tfevents.1717635033.4a1f19113fa6

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ee45b727ef73050650719d6370bb442ed773c108c5efa37345fd1a1195f4cae0
+size 2433024

.summary/0/events.out.tfevents.1717700366.c11948e043d9

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1cf6cafad438ea9510a7e6f67df1fb9eb1ff981d02fe2e588bfa5af45242fe1e
+size 861005

.summary/0/events.out.tfevents.1717702446.c11948e043d9

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8603f0d9093699a29995f4ae331c45933b2d8ebc4a2f416da7a29df9b07ab809
+size 4371047

.summary/0/events.out.tfevents.1717708092.c11948e043d9

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ddeda34714771693362020b4a67d629000665abf6347166f03b8b5b7d43765c3
+size 40

.summary/0/events.out.tfevents.1717708590.c11948e043d9

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d5918fd3155796d72554d67c6aa10370132f2c23daf769ed383de0d09a63f693
+size 1183336

.summary/0/events.out.tfevents.1717710273.c11948e043d9

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f9de5828b5e679da8fd5921ac0b5419e928d328aabb81b0e0d59e7eac9370a41
+size 823211

.summary/0/events.out.tfevents.1717711462.c11948e043d9

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:842dd39fceac52fe91d70e09956b210b7a0dd3034e287fa88caa6d8d65e982af
+size 3756481

README.md

@@ -0,0 +1,42 @@
+---
+library_name: sample-factory
+tags:
+- deep-reinforcement-learning
+- reinforcement-learning
+- sample-factory
+---
+
+A(n) **APPO** model trained on the **GDY-MettaGrid** environment.
+
+This model was trained using Sample-Factory 2.0: https://github.com/alex-petrenko/sample-factory.
+Documentation for how to use Sample-Factory can be found at https://www.samplefactory.dev/
+
+
+## Downloading the model
+
+After installing Sample-Factory, download the model with:
+```
+python -m sample_factory.huggingface.load_from_hub -r metta-ai/baseline.v0.2.0
+```
+
+    
+## Using the model
+
+To run the model after download, use the `enjoy` script corresponding to this environment:
+```
+python -m <path.to.enjoy.module> --algo=APPO --env=GDY-MettaGrid --train_dir=./train_dir --experiment=baseline.v0.2.0
+```
+
+
+You can also upload models to the Hugging Face Hub using the same script with the `--push_to_hub` flag.
+See https://www.samplefactory.dev/10-huggingface/huggingface/ for more details
+    
+## Training with this model
+
+To continue training with this model, use the `train` script corresponding to this environment:
+```
+python -m <path.to.train.module> --algo=APPO --env=GDY-MettaGrid --train_dir=./train_dir --experiment=baseline.v0.2.0 --restart_behavior=resume --train_for_env_steps=10000000000
+```
+
+Note, you may have to adjust `--train_for_env_steps` to a suitably high number as the experiment will resume at the number of steps it concluded at.
+    

a100_100x100.mp4

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dea77db32fda2241d5541d48313bdedb572bcbb994004f10b4b723e205e2e203
+size 9661669

a20_40x40.mp4

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7f89d9f3d72b2382ffeffab31cd7921e8907f93413a6ccad5131cf50821db2f4
+size 2323340

a20_b4_40x40.mp4

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c2dcf30fbf2adad4991011cfa642c43fcb75f1355eb481f35f1a7b332fa87dea
+size 2240752

a20_r4_40x40.mp4

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:640d7928519f980324a62f09cb517eba01feb8e431595a403030fe9821a76446
+size 2733835

checkpoint_p0/best_000028313_463880192_reward_0.311.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f54f6415da4cc53fad8860311d0f098923f8b2ea998245c9c6d409ae3aca73cb
+size 38973114

checkpoint_p0/checkpoint_000045196_740491264.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:64334522b515ed267d3bfb0f1a6dab8ee7b55a5f67b4511a6143a5c655eac0f1
+size 38974422

checkpoint_p0/checkpoint_000045522_745832448.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:13ceaa7a747668f639814f25e3f6c02f80b040affdd8aa6d89b36c21d8dc965b
+size 38974422

config.json

@@ -0,0 +1,161 @@
+{
+  "help": false,
+  "algo": "APPO",
+  "env": "GDY-MettaGrid",
+  "experiment": "p2.metta.4",
+  "train_dir": "/workspace/metta/train_dir",
+  "restart_behavior": "resume",
+  "device": "gpu",
+  "seed": 0,
+  "num_policies": 1,
+  "async_rl": true,
+  "serial_mode": false,
+  "batched_sampling": false,
+  "num_batches_to_accumulate": 2,
+  "worker_num_splits": 2,
+  "policy_workers_per_policy": 1,
+  "max_policy_lag": 50,
+  "num_workers": 32,
+  "num_envs_per_worker": 2,
+  "batch_size": 16384,
+  "num_batches_per_epoch": 1,
+  "num_epochs": 1,
+  "rollout": 256,
+  "recurrence": 256,
+  "shuffle_minibatches": false,
+  "gamma": 0.99,
+  "reward_scale": 1.0,
+  "reward_clip": 1000.0,
+  "value_bootstrap": false,
+  "normalize_returns": true,
+  "exploration_loss_coeff": 0.002,
+  "value_loss_coeff": 0.976,
+  "kl_loss_coeff": 0.0,
+  "aux_loss_coeff": 0.0,
+  "exploration_loss": "symmetric_kl",
+  "gae_lambda": 0.95,
+  "ppo_clip_ratio": 0.1,
+  "ppo_clip_value": 1.0,
+  "with_vtrace": false,
+  "vtrace_rho": 1.0,
+  "vtrace_c": 1.0,
+  "optimizer": "adam",
+  "adam_eps": 1e-06,
+  "adam_beta1": 0.9,
+  "adam_beta2": 0.999,
+  "max_grad_norm": 4.0,
+  "learning_rate": 0.0001,
+  "lr_schedule": "constant",
+  "lr_schedule_kl_threshold": 0.008,
+  "lr_adaptive_min": 1e-06,
+  "lr_adaptive_max": 0.01,
+  "obs_subtract_mean": 0.0,
+  "obs_scale": 1.0,
+  "normalize_input": false,
+  "normalize_input_keys": null,
+  "decorrelate_experience_max_seconds": 150,
+  "decorrelate_envs_on_one_worker": true,
+  "actor_worker_gpus": [],
+  "set_workers_cpu_affinity": true,
+  "force_envs_single_thread": false,
+  "default_niceness": 0,
+  "log_to_file": true,
+  "experiment_summaries_interval": 10,
+  "flush_summaries_interval": 30,
+  "stats_avg": 100,
+  "summaries_use_frameskip": true,
+  "heartbeat_interval": 20,
+  "heartbeat_reporting_interval": 180,
+  "train_for_env_steps": 9999999999999,
+  "train_for_seconds": 10000000000,
+  "save_every_sec": 120,
+  "keep_checkpoints": 2,
+  "load_checkpoint_path": null,
+  "init_checkpoint_path": null,
+  "load_checkpoint_kind": "latest",
+  "save_milestones_sec": -1,
+  "save_best_every_sec": 5,
+  "save_best_metric": "reward",
+  "save_best_after": 100000,
+  "benchmark": false,
+  "encoder_mlp_layers": [
+    512,
+    512
+  ],
+  "encoder_conv_architecture": "convnet_simple",
+  "encoder_conv_mlp_layers": [
+    512
+  ],
+  "use_rnn": true,
+  "rnn_size": 512,
+  "rnn_type": "gru",
+  "rnn_num_layers": 1,
+  "decoder_mlp_layers": [],
+  "nonlinearity": "elu",
+  "policy_initialization": "orthogonal",
+  "policy_init_gain": 1.0,
+  "actor_critic_share_weights": true,
+  "adaptive_stddev": true,
+  "continuous_tanh_scale": 0.0,
+  "initial_stddev": 1.0,
+  "use_env_info_cache": false,
+  "env_gpu_actions": false,
+  "env_gpu_observations": true,
+  "env_frameskip": 1,
+  "env_framestack": 1,
+  "pixel_format": "CHW",
+  "use_record_episode_statistics": false,
+  "episode_counter": false,
+  "with_wandb": true,
+  "wandb_user": "platypus",
+  "wandb_project": "metta",
+  "wandb_group": null,
+  "wandb_job_type": "SF",
+  "wandb_tags": [],
+  "with_pbt": false,
+  "pbt_mix_policies_in_one_env": true,
+  "pbt_period_env_steps": 5000000,
+  "pbt_start_mutation": 20000000,
+  "pbt_replace_fraction": 0.3,
+  "pbt_mutation_rate": 0.15,
+  "pbt_replace_reward_gap": 0.1,
+  "pbt_replace_reward_gap_absolute": 1e-06,
+  "pbt_optimize_gamma": false,
+  "pbt_target_objective": "true_objective",
+  "pbt_perturb_min": 1.1,
+  "pbt_perturb_max": 1.5,
+  "env_cfg": "{\"name\": \"GDY-MettaGrid\", \"_target_\": \"env.griddly.mettagrid.gym_env.MettaGridGymEnv\", \"max_action_value\": 10, \"game_builder\": {\"_target_\": \"env.griddly.mettagrid.game_builder.MettaGridGameBuilder\", \"obs_width\": 11, \"obs_height\": 11, \"max_steps\": 1000, \"tile_size\": 16, \"num_agents\": 20, \"objects\": {\"agent\": {\"initial_energy\": 100, \"max_energy\": 500, \"max_inventory\": 5, \"freeze_duration\": 10, \"hp\": 1, \"upkeep\": {\"shield\": 1}}, \"altar\": {\"hp\": 30, \"cooldown\": 2, \"cost\": 100}, \"converter\": {\"hp\": 30, \"cooldown\": 2, \"energy_output\": 50, \"cost\": 0}, \"generator\": {\"hp\": 30, \"cooldown\": 50, \"initial_resources\": 100, \"cost\": 0}, \"wall\": {\"density\": 0.01, \"hp\": 10}}, \"actions\": {\"move\": {\"cost\": 0}, \"rotate\": {\"cost\": 0}, \"jump\": {\"cost\": 3}, \"shield\": {\"cost\": 0}, \"drop\": {\"cost\": 1}, \"use\": {\"cost\": 0}, \"attack\": {\"cost\": 5, \"damage\": 5}}, \"map\": {\"layout\": [[\"base\", \"wild_1\", \"base\"], [\"wild_2\", \"center\", \"wild_2\"], [\"base\", \"wild_1\", \"base\"]], \"room\": {\"width\": 25, \"height\": 25, \"num_agents\": 5, \"objects\": {\"agent\": 5, \"altar\": 1, \"converter\": 3, \"generator\": 15, \"wall\": 10}}, \"wild_1\": {\"width\": 10, \"height\": 15, \"border\": 0, \"objects\": {\"agent\": 0, \"altar\": 1, \"converter\": 1, \"generator\": 5, \"wall\": 5}}, \"wild_2\": {\"width\": 15, \"height\": 10, \"border\": 0, \"objects\": {\"agent\": 0, \"altar\": 1, \"converter\": 1, \"generator\": 5, \"wall\": 5}}, \"center\": {\"width\": 10, \"height\": 10, \"border\": 0, \"objects\": {\"agent\": 0, \"altar\": 2, \"converter\": 5, \"generator\": 10, \"wall\": 20}}, \"base\": {\"width\": 15, \"height\": 15, \"border\": 1, \"objects\": {\"agent\": 5, \"altar\": 1, \"converter\": 3, \"generator\": 5, \"wall\": 5}}}}}",
+  "agent_cfg": "{\"_target_\": \"agent.metta_agent.MettaAgent\", \"observation_encoders\": {\"grid_obs\": {\"feature_names\": [], \"normalize_features\": true, \"label_dim\": 4, \"output_dim\": 512, \"layers\": 4}, \"global_vars\": {\"feature_names\": [], \"normalize_features\": true, \"label_dim\": 4, \"output_dim\": 8, \"layers\": 2}, \"last_action\": {\"feature_names\": [\"last_action_id\", \"last_action_val\"], \"normalize_features\": true, \"label_dim\": 4, \"output_dim\": 8, \"layers\": 2}, \"last_reward\": {\"feature_names\": [\"last_reward\"], \"normalize_features\": true, \"label_dim\": 4, \"output_dim\": 8, \"layers\": 2}}, \"fc\": {\"layers\": 3, \"output_dim\": 512}, \"decoder\": {\"_target_\": \"agent.decoder.Decoder\"}, \"core\": {\"_target_\": \"sample_factory.model.core.ModelCoreRNN\", \"rnn_type\": \"gru\", \"rnn_num_layers\": 1, \"rnn_size\": 512}}",
+  "command_line": "--aux_loss_coef=0 --recurrence=256 --rollout=256 --value_loss_coeff=0.976 --exploration_loss=symmetric_kl --exploration_loss_coeff=0.002 --policy_initialization=orthogonal --learning_rate=0.0001 --max_policy_lag=50 --nonlinearity=elu --load_checkpoint_kind=latest --normalize_input=False --seed=0 --batch_size=16384 --decorrelate_experience_max_seconds=150 --train_for_env_steps=9999999999999 --with_wandb=True --wandb_user=platypus --wandb_project=metta --experiment=p2.metta.4 --rnn_type=gru --rnn_num_layers=1 --rnn_size=512 --env=GDY-MettaGrid --env_cfg={\"name\": \"GDY-MettaGrid\", \"_target_\": \"env.griddly.mettagrid.gym_env.MettaGridGymEnv\", \"game_builder\": {\"_target_\": \"env.griddly.mettagrid.game_builder.MettaGridGameBuilder\", \"obs_width\": 11, \"obs_height\": 11, \"max_steps\": 1000, \"tile_size\": 16, \"width\": 40, \"height\": 40, \"objects\": {\"agent\": {\"count\": 20, \"initial_energy\": 100, \"max_energy\": 500, \"max_inventory\": 5, \"freeze_duration\": 10, \"hp\": 1, \"upkeep\": {\"shield\": 1}}, \"altar\": {\"hp\": 30, \"count\": 5, \"cooldown\": 2, \"cost\": 100}, \"converter\": {\"hp\": 30, \"count\": 10, \"cooldown\": 2, \"energy_output\": 50, \"cost\": 0}, \"generator\": {\"hp\": 30, \"count\": 50, \"cooldown\": 50, \"initial_resources\": 100, \"cost\": 0}, \"wall\": {\"density\": 0.01, \"hp\": 10}, \"charger\": {\"count\": 10}}, \"actions\": {\"move\": {\"cost\": 0}, \"rotate\": {\"cost\": 0}, \"jump\": {\"cost\": 3}, \"shield\": {\"cost\": 0}, \"drop\": {\"cost\": 1}, \"use\": {\"cost\": 0}, \"attack\": {\"cost\": 5, \"damage\": 5}}}, \"global_features\": [\"conf:agent:energy:initial\", \"conf:agent:energy:max\", \"conf:altar:cooldown\", \"conf:altar:cost\", \"conf:attack:damage\", \"conf:attack:freeze_duration\", \"conf:charger:cooldown\", \"conf:charger:energy\", \"conf:cost:attack\", \"conf:cost:frozen\", \"conf:cost:jump\", \"conf:cost:move\", \"conf:cost:rotate\", \"conf:cost:shield:upkeep\", \"conf:cost:shield\", \"conf:generator:cooldown\", \"conf:gift:energy\"], \"grid_features\": [\"agent\", \"altar\", \"charger\", \"generator\", \"wall\", \"agent:dir\", \"agent:energy\", \"agent:frozen\", \"agent:id\", \"agent:inv:1\", \"agent:inv:2\", \"agent:inv:3\", \"agent:shield\", \"agent:species\", \"altar:ready\", \"charger:bonus\", \"charger:input:1\", \"charger:input:2\", \"charger:input:3\", \"charger:output\", \"charger:ready\", \"generator:ready\", \"generator:resource\"], \"actions\": [\"rotate\", \"move\", \"jump\", \"use\", \"gift\", \"shield\", \"attack\"], \"max_action_value\": 10} --agent_cfg={\"_target_\": \"agent.metta_agent.MettaAgent\", \"observation_encoders\": {\"grid_obs\": {\"feature_names\": [], \"normalize_features\": true, \"label_dim\": 4, \"output_dim\": 512, \"layers\": 4}, \"global_vars\": {\"feature_names\": [], \"normalize_features\": true, \"label_dim\": 4, \"output_dim\": 8, \"layers\": 2}, \"last_action\": {\"feature_names\": [\"last_action_id\", \"last_action_val\"], \"normalize_features\": true, \"label_dim\": 4, \"output_dim\": 8, \"layers\": 2}, \"last_reward\": {\"feature_names\": [\"last_reward\"], \"normalize_features\": true, \"label_dim\": 4, \"output_dim\": 8, \"layers\": 2}}, \"fc\": {\"layers\": 3, \"output_dim\": 512}, \"decoder\": {\"_target_\": \"agent.decoder.Decoder\"}, \"core\": {\"_target_\": \"sample_factory.model.core.ModelCoreRNN\", \"rnn_type\": \"gru\", \"rnn_num_layers\": 1, \"rnn_size\": 512}}",
+  "cli_args": {
+    "env": "GDY-MettaGrid",
+    "experiment": "p2.metta.4",
+    "seed": 0,
+    "max_policy_lag": 50,
+    "batch_size": 16384,
+    "rollout": 256,
+    "recurrence": 256,
+    "exploration_loss_coeff": 0.002,
+    "value_loss_coeff": 0.976,
+    "aux_loss_coeff": 0.0,
+    "exploration_loss": "symmetric_kl",
+    "learning_rate": 0.0001,
+    "normalize_input": false,
+    "decorrelate_experience_max_seconds": 150,
+    "train_for_env_steps": 9999999999999,
+    "load_checkpoint_kind": "latest",
+    "rnn_size": 512,
+    "rnn_type": "gru",
+    "rnn_num_layers": 1,
+    "nonlinearity": "elu",
+    "policy_initialization": "orthogonal",
+    "with_wandb": true,
+    "wandb_user": "platypus",
+    "wandb_project": "metta",
+    "env_cfg": "{\"name\": \"GDY-MettaGrid\", \"_target_\": \"env.griddly.mettagrid.gym_env.MettaGridGymEnv\", \"game_builder\": {\"_target_\": \"env.griddly.mettagrid.game_builder.MettaGridGameBuilder\", \"obs_width\": 11, \"obs_height\": 11, \"max_steps\": 1000, \"tile_size\": 16, \"width\": 40, \"height\": 40, \"objects\": {\"agent\": {\"count\": 20, \"initial_energy\": 100, \"max_energy\": 500, \"max_inventory\": 5, \"freeze_duration\": 10, \"hp\": 1, \"upkeep\": {\"shield\": 1}}, \"altar\": {\"hp\": 30, \"count\": 5, \"cooldown\": 2, \"cost\": 100}, \"converter\": {\"hp\": 30, \"count\": 10, \"cooldown\": 2, \"energy_output\": 50, \"cost\": 0}, \"generator\": {\"hp\": 30, \"count\": 50, \"cooldown\": 50, \"initial_resources\": 100, \"cost\": 0}, \"wall\": {\"density\": 0.01, \"hp\": 10}, \"charger\": {\"count\": 10}}, \"actions\": {\"move\": {\"cost\": 0}, \"rotate\": {\"cost\": 0}, \"jump\": {\"cost\": 3}, \"shield\": {\"cost\": 0}, \"drop\": {\"cost\": 1}, \"use\": {\"cost\": 0}, \"attack\": {\"cost\": 5, \"damage\": 5}}}, \"global_features\": [\"conf:agent:energy:initial\", \"conf:agent:energy:max\", \"conf:altar:cooldown\", \"conf:altar:cost\", \"conf:attack:damage\", \"conf:attack:freeze_duration\", \"conf:charger:cooldown\", \"conf:charger:energy\", \"conf:cost:attack\", \"conf:cost:frozen\", \"conf:cost:jump\", \"conf:cost:move\", \"conf:cost:rotate\", \"conf:cost:shield:upkeep\", \"conf:cost:shield\", \"conf:generator:cooldown\", \"conf:gift:energy\"], \"grid_features\": [\"agent\", \"altar\", \"charger\", \"generator\", \"wall\", \"agent:dir\", \"agent:energy\", \"agent:frozen\", \"agent:id\", \"agent:inv:1\", \"agent:inv:2\", \"agent:inv:3\", \"agent:shield\", \"agent:species\", \"altar:ready\", \"charger:bonus\", \"charger:input:1\", \"charger:input:2\", \"charger:input:3\", \"charger:output\", \"charger:ready\", \"generator:ready\", \"generator:resource\"], \"actions\": [\"rotate\", \"move\", \"jump\", \"use\", \"gift\", \"shield\", \"attack\"], \"max_action_value\": 10}",
+    "agent_cfg": "{\"_target_\": \"agent.metta_agent.MettaAgent\", \"observation_encoders\": {\"grid_obs\": {\"feature_names\": [], \"normalize_features\": true, \"label_dim\": 4, \"output_dim\": 512, \"layers\": 4}, \"global_vars\": {\"feature_names\": [], \"normalize_features\": true, \"label_dim\": 4, \"output_dim\": 8, \"layers\": 2}, \"last_action\": {\"feature_names\": [\"last_action_id\", \"last_action_val\"], \"normalize_features\": true, \"label_dim\": 4, \"output_dim\": 8, \"layers\": 2}, \"last_reward\": {\"feature_names\": [\"last_reward\"], \"normalize_features\": true, \"label_dim\": 4, \"output_dim\": 8, \"layers\": 2}}, \"fc\": {\"layers\": 3, \"output_dim\": 512}, \"decoder\": {\"_target_\": \"agent.decoder.Decoder\"}, \"core\": {\"_target_\": \"sample_factory.model.core.ModelCoreRNN\", \"rnn_type\": \"gru\", \"rnn_num_layers\": 1, \"rnn_size\": 512}}"
+  },
+  "git_hash": "7864b034e8faa45a8cef257cdd0b4d3b2b2549f3",
+  "git_repo_name": "https://github.com/daveey/metta.git",
+  "wandb_unique_id": "p2.metta.4_20240605_175031_058004"
+}

replay.mp4

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7f89d9f3d72b2382ffeffab31cd7921e8907f93413a6ccad5131cf50821db2f4
+size 2323340