Upload folder using huggingface_hub

.summary/0/events.out.tfevents.1697015803.rhmmedcatt-proliant-ml350-gen10

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

.summary/1/events.out.tfevents.1697015803.rhmmedcatt-proliant-ml350-gen10

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

README.md

@@ -15,7 +15,7 @@ model-index:
       type: atari_freeway
     metrics:
     - type: mean_reward
-      value: 0.00 +/- 0.00
+      value: 34.00 +/- 0.00
       name: mean_reward
       verified: false
 ---
@@ -30,15 +30,128 @@ Documentation for how to use Sample-Factory can be found at https://www.samplefa
 
 After installing Sample-Factory, download the model with:
 ```
-python -m sample_factory.huggingface.load_from_hub -r MattStammers/appo-atari_freeway
+python -m sample_factory.huggingface.load_from_hub -r MattStammers/APPO-atari_freeway
 ```
 
     
+## About the Model
+
+This model as with all the others in the benchmarks was trained initially asynchronously un-seeded to 10 million steps for the purposes of setting a sample factory async baseline for this model on this environment but only 3/57 made it. 
+
+The aim is to reach state-of-the-art (SOTA) performance on each atari environment. I will flag the models with SOTA when they reach at or near these levels. 
+
+The hyperparameters used in the model are the ones I have pushed to my fork of sample-factory: https://github.com/MattStammers/sample-factory. Given that https://huggingface.co/edbeeching has kindly shared his.
+I saved time and energy by using many of his tuned hyperparameters to maximise performance. However, he used 2 billion training steps. I have started as explained above at 10 million then moved to 100m to see how performance goes:
+```
+hyperparameters =  {
+  "device": "gpu",
+  "seed": 1234,
+  "num_policies": 2,
+  "async_rl": true,
+  "serial_mode": false,
+  "batched_sampling": true,
+  "num_batches_to_accumulate": 2,
+  "worker_num_splits": 1,
+  "policy_workers_per_policy": 1,
+  "max_policy_lag": 1000,
+  "num_workers": 16,
+  "num_envs_per_worker": 2,
+  "batch_size": 1024,
+  "num_batches_per_epoch": 8,
+  "num_epochs": 4,
+  "rollout": 128,
+  "recurrence": 1,
+  "shuffle_minibatches": false,
+  "gamma": 0.99,
+  "reward_scale": 1.0,
+  "reward_clip": 1000.0,
+  "value_bootstrap": false,
+  "normalize_returns": true,
+  "exploration_loss_coeff": 0.0004677351413,
+  "value_loss_coeff": 0.5,
+  "kl_loss_coeff": 0.0,
+  "exploration_loss": "entropy",
+  "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-05,
+  "adam_beta1": 0.9,
+  "adam_beta2": 0.999,
+  "max_grad_norm": 0.0,
+  "learning_rate": 0.0003033891184,
+  "lr_schedule": "linear_decay",
+  "lr_schedule_kl_threshold": 0.008,
+  "lr_adaptive_min": 1e-06,
+  "lr_adaptive_max": 0.01,
+  "obs_subtract_mean": 0.0,
+  "obs_scale": 255.0,
+  "normalize_input": true,
+  "normalize_input_keys": [
+    "obs"
+  ],
+  "decorrelate_experience_max_seconds": 0,
+  "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": 3,
+  "flush_summaries_interval": 30,
+  "stats_avg": 100,
+  "summaries_use_frameskip": true,
+  "heartbeat_interval": 10,
+  "heartbeat_reporting_interval": 60,
+  "train_for_env_steps": 100000000,
+  "train_for_seconds": 10000000000,
+  "save_every_sec": 120,
+  "keep_checkpoints": 2,
+  "load_checkpoint_kind": "latest",
+  "save_milestones_sec": 1200,
+  "save_best_every_sec": 5,
+  "save_best_metric": "reward",
+  "save_best_after": 100000,
+  "benchmark": false,
+  "encoder_mlp_layers": [
+    512,
+    512
+  ],
+  "encoder_conv_architecture": "convnet_atari",
+  "encoder_conv_mlp_layers": [
+    512
+  ],
+  "use_rnn": false,
+  "rnn_size": 512,
+  "rnn_type": "gru",
+  "rnn_num_layers": 1,
+  "decoder_mlp_layers": [],
+  "nonlinearity": "relu",
+  "policy_initialization": "orthogonal",
+  "policy_init_gain": 1.0,
+  "actor_critic_share_weights": true,
+  "adaptive_stddev": false,
+  "continuous_tanh_scale": 0.0,
+  "initial_stddev": 1.0,
+  "use_env_info_cache": false,
+  "env_gpu_actions": false,
+  "env_gpu_observations": true,
+  "env_frameskip": 4,
+  "env_framestack": 4,
+  }
+
+  ```
+
+
+    
 ## Using the model
 
 To run the model after download, use the `enjoy` script corresponding to this environment:
 ```
-python -m sf_examples.atari.enjoy_atari --algo=APPO --env=atari_freeway --train_dir=./train_dir --experiment=appo-atari_freeway
+python -m sf_examples.atari.enjoy_atari --algo=APPO --env=atari_freeway --train_dir=./train_dir --experiment=APPO-atari_freeway
 ```
 
 
@@ -49,7 +162,7 @@ See https://www.samplefactory.dev/10-huggingface/huggingface/ for more details
 
 To continue training with this model, use the `train` script corresponding to this environment:
 ```
-python -m sf_examples.atari.train_atari --algo=APPO --env=atari_freeway --train_dir=./train_dir --experiment=appo-atari_freeway --restart_behavior=resume --train_for_env_steps=10000000000
+python -m sf_examples.atari.train_atari --algo=APPO --env=atari_freeway --train_dir=./train_dir --experiment=APPO-atari_freeway --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.

checkpoint_p0/best_000089472_91619328_reward_34.000.pth

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

checkpoint_p0/checkpoint_000097216_99549184.pth

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

checkpoint_p0/checkpoint_000097664_100007936.pth

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

checkpoint_p0/milestones/checkpoint_000007104_7274496.pth

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

checkpoint_p0/milestones/checkpoint_000014304_14647296.pth

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

checkpoint_p0/milestones/checkpoint_000021504_22020096.pth

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

checkpoint_p0/milestones/checkpoint_000028736_29425664.pth

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

checkpoint_p0/milestones/checkpoint_000035936_36798464.pth

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

checkpoint_p0/milestones/checkpoint_000043168_44204032.pth

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

checkpoint_p0/milestones/checkpoint_000050368_51576832.pth

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

checkpoint_p0/milestones/checkpoint_000057568_58949632.pth

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

checkpoint_p0/milestones/checkpoint_000064768_66322432.pth

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

checkpoint_p0/milestones/checkpoint_000071968_73695232.pth

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

checkpoint_p0/milestones/checkpoint_000079200_81100800.pth

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

checkpoint_p0/milestones/checkpoint_000086432_88506368.pth

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

checkpoint_p0/milestones/checkpoint_000093632_95879168.pth

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

checkpoint_p1/best_000097280_99614720_reward_34.000.pth

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

checkpoint_p1/checkpoint_000097792_100139008.pth

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

checkpoint_p1/checkpoint_000098272_100630528.pth

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

checkpoint_p1/milestones/checkpoint_000007136_7307264.pth

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

checkpoint_p1/milestones/checkpoint_000014400_14745600.pth

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

checkpoint_p1/milestones/checkpoint_000021664_22183936.pth

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

checkpoint_p1/milestones/checkpoint_000028896_29589504.pth

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

checkpoint_p1/milestones/checkpoint_000036160_37027840.pth

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

checkpoint_p1/milestones/checkpoint_000043392_44433408.pth

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

checkpoint_p1/milestones/checkpoint_000050656_51871744.pth

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

checkpoint_p1/milestones/checkpoint_000057920_59310080.pth

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

checkpoint_p1/milestones/checkpoint_000065152_66715648.pth

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

checkpoint_p1/milestones/checkpoint_000072416_74153984.pth

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

checkpoint_p1/milestones/checkpoint_000079680_81592320.pth

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

checkpoint_p1/milestones/checkpoint_000086912_88997888.pth

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

checkpoint_p1/milestones/checkpoint_000094176_96436224.pth

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

config.json

@@ -2,23 +2,23 @@
   "help": false,
   "algo": "APPO",
   "env": "atari_freeway",
-  "experiment": "atari_freeway",
+  "experiment": "atari_freeway_APPO",
   "train_dir": "./train_atari",
-  "restart_behavior": "resume",
+  "restart_behavior": "restart",
   "device": "gpu",
-  "seed": null,
+  "seed": 1234,
   "num_policies": 2,
-  "async_rl": false,
+  "async_rl": true,
   "serial_mode": false,
-  "batched_sampling": false,
+  "batched_sampling": true,
   "num_batches_to_accumulate": 2,
   "worker_num_splits": 1,
   "policy_workers_per_policy": 1,
   "max_policy_lag": 1000,
-  "num_workers": 8,
-  "num_envs_per_worker": 1,
-  "batch_size": 256,
-  "num_batches_per_epoch": 4,
+  "num_workers": 16,
+  "num_envs_per_worker": 2,
+  "batch_size": 1024,
+  "num_batches_per_epoch": 8,
   "num_epochs": 4,
   "rollout": 128,
   "recurrence": 1,
@@ -28,7 +28,7 @@
   "reward_clip": 1000.0,
   "value_bootstrap": false,
   "normalize_returns": true,
-  "exploration_loss_coeff": 0.01,
+  "exploration_loss_coeff": 0.0004677351413,
   "value_loss_coeff": 0.5,
   "kl_loss_coeff": 0.0,
   "exploration_loss": "entropy",
@@ -42,8 +42,8 @@
   "adam_eps": 1e-05,
   "adam_beta1": 0.9,
   "adam_beta2": 0.999,
-  "max_grad_norm": 0.5,
-  "learning_rate": 0.00025,
+  "max_grad_norm": 0.0,
+  "learning_rate": 0.0003033891184,
   "lr_schedule": "linear_decay",
   "lr_schedule_kl_threshold": 0.008,
   "lr_adaptive_min": 1e-06,
@@ -51,7 +51,9 @@
   "obs_subtract_mean": 0.0,
   "obs_scale": 255.0,
   "normalize_input": true,
-  "normalize_input_keys": null,
+  "normalize_input_keys": [
+    "obs"
+  ],
   "decorrelate_experience_max_seconds": 0,
   "decorrelate_envs_on_one_worker": true,
   "actor_worker_gpus": [],
@@ -63,14 +65,14 @@
   "flush_summaries_interval": 30,
   "stats_avg": 100,
   "summaries_use_frameskip": true,
-  "heartbeat_interval": 20,
-  "heartbeat_reporting_interval": 180,
-  "train_for_env_steps": 10000000,
+  "heartbeat_interval": 10,
+  "heartbeat_reporting_interval": 60,
+  "train_for_env_steps": 100000000,
   "train_for_seconds": 10000000000,
   "save_every_sec": 120,
   "keep_checkpoints": 2,
   "load_checkpoint_kind": "latest",
-  "save_milestones_sec": -1,
+  "save_milestones_sec": 1200,
   "save_best_every_sec": 5,
   "save_best_metric": "reward",
   "save_best_after": 100000,
@@ -104,7 +106,7 @@
   "use_record_episode_statistics": true,
   "with_wandb": true,
   "wandb_user": "matt-stammers",
-  "wandb_project": "atari",
+  "wandb_project": "atari_APPO",
   "wandb_group": "atari_freeway",
   "wandb_job_type": "SF",
   "wandb_tags": [
@@ -122,16 +124,32 @@
   "pbt_target_objective": "true_objective",
   "pbt_perturb_min": 1.1,
   "pbt_perturb_max": 1.5,
-  "command_line": "--algo=APPO --env=atari_freeway --experiment=atari_freeway --num_policies=2 --train_dir=./train_atari --with_wandb=true --wandb_user=matt-stammers --wandb_project=atari --wandb_group=atari_freeway --wandb_job_type=SF --wandb_tags=atari",
+  "command_line": "--algo=APPO --env=atari_freeway --experiment=atari_freeway_APPO --num_policies=2 --restart_behavior=restart --train_dir=./train_atari --train_for_env_steps=100000000 --seed=1234 --num_workers=16 --num_envs_per_worker=2 --num_batches_per_epoch=8 --async_rl=true --batched_sampling=true --batch_size=1024 --max_grad_norm=0 --learning_rate=0.0003033891184 --heartbeat_interval=10 --heartbeat_reporting_interval=60 --save_milestones_sec=1200 --num_epochs=4 --exploration_loss_coeff=0.0004677351413 --with_wandb=true --wandb_user=matt-stammers --wandb_project=atari_APPO --wandb_group=atari_freeway --wandb_job_type=SF --wandb_tags=atari",
   "cli_args": {
     "algo": "APPO",
     "env": "atari_freeway",
-    "experiment": "atari_freeway",
+    "experiment": "atari_freeway_APPO",
     "train_dir": "./train_atari",
+    "restart_behavior": "restart",
+    "seed": 1234,
     "num_policies": 2,
+    "async_rl": true,
+    "batched_sampling": true,
+    "num_workers": 16,
+    "num_envs_per_worker": 2,
+    "batch_size": 1024,
+    "num_batches_per_epoch": 8,
+    "num_epochs": 4,
+    "exploration_loss_coeff": 0.0004677351413,
+    "max_grad_norm": 0.0,
+    "learning_rate": 0.0003033891184,
+    "heartbeat_interval": 10,
+    "heartbeat_reporting_interval": 60,
+    "train_for_env_steps": 100000000,
+    "save_milestones_sec": 1200,
     "with_wandb": true,
     "wandb_user": "matt-stammers",
-    "wandb_project": "atari",
+    "wandb_project": "atari_APPO",
     "wandb_group": "atari_freeway",
     "wandb_job_type": "SF",
     "wandb_tags": [
@@ -140,5 +158,5 @@
   },
   "git_hash": "5fff97c2f535da5987d358cdbe6927cccd43621e",
   "git_repo_name": "not a git repository",
-  "wandb_unique_id": "atari_freeway_20230926_064708_504956"
+  "wandb_unique_id": "atari_freeway_APPO_20231011_101641_307643"
 }

git.diff

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:28f3fe545f35e5db1e419bf1e198c0bc5ef89459d55f59e2d06d3c108907e732
-size 14403396
+oid sha256:3357904f421d3f4924836316b1741bf64d5dd0e807d5e80ac07059b4c52a7008
+size 14426734

replay.mp4

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:2d75add18affe8ba78d67a99984b6bce52857e16def6bba597d7ebfeef244030
-size 6792275
+oid sha256:fe24b246f3e9b2d760424f65a4c3860f17bf503131d8e8ae8bab07ce1735597d
+size 6947894