Hugging Face's logo

eloialonso
/
diamond

Reinforcement Learning
diffusion
world-models
Model card Files Community
2
eloialonso commited on
Commit
7ecb82e
•
1 Parent(s): 65c5de2

Upload folder using huggingface_hub

Browse files
Files changed (28) hide show
  1. atari_100k/config/agent/default.yaml +33 -0
  2. atari_100k/config/env/atari.yaml +13 -0
  3. atari_100k/models/Alien.pt +3 -0
  4. atari_100k/models/Amidar.pt +3 -0
  5. atari_100k/models/Assault.pt +3 -0
  6. atari_100k/models/Asterix.pt +3 -0
  7. atari_100k/models/BankHeist.pt +3 -0
  8. atari_100k/models/BattleZone.pt +3 -0
  9. atari_100k/models/Boxing.pt +3 -0
  10. atari_100k/models/Breakout.pt +3 -0
  11. atari_100k/models/ChopperCommand.pt +3 -0
  12. atari_100k/models/CrazyClimber.pt +3 -0
  13. atari_100k/models/DemonAttack.pt +3 -0
  14. atari_100k/models/Freeway.pt +3 -0
  15. atari_100k/models/Frostbite.pt +3 -0
  16. atari_100k/models/Gopher.pt +3 -0
  17. atari_100k/models/Hero.pt +3 -0
  18. atari_100k/models/Jamesbond.pt +3 -0
  19. atari_100k/models/Kangaroo.pt +3 -0
  20. atari_100k/models/Krull.pt +3 -0
  21. atari_100k/models/KungFuMaster.pt +3 -0
  22. atari_100k/models/MsPacman.pt +3 -0
  23. atari_100k/models/Pong.pt +3 -0
  24. atari_100k/models/PrivateEye.pt +3 -0
  25. atari_100k/models/Qbert.pt +3 -0
  26. atari_100k/models/RoadRunner.pt +3 -0
  27. atari_100k/models/Seaquest.pt +3 -0
  28. atari_100k/models/UpNDown.pt +3 -0
atari_100k/config/agent/default.yaml ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ _target_: agent.AgentConfig
2
+
3
+ denoiser:
4
+ _target_: models.diffusion.DenoiserConfig
5
+ sigma_data: 0.5
6
+ sigma_offset_noise: 0.3
7
+ inner_model:
8
+ _target_: models.diffusion.InnerModelConfig
9
+ img_channels: 3
10
+ num_steps_conditioning: 4
11
+ cond_channels: 256
12
+ depths: [2,2,2,2]
13
+ channels: [64,64,64,64]
14
+ attn_depths: [0,0,0,0]
15
+
16
+ rew_end_model:
17
+ _target_: models.rew_end_model.RewEndModelConfig
18
+ lstm_dim: 512
19
+ img_channels: ${agent.denoiser.inner_model.img_channels}
20
+ img_size: ${env.train.size}
21
+ cond_channels: 128
22
+ depths: [2,2,2,2]
23
+ channels: [32,32,32,32]
24
+ attn_depths: [0,0,0,0]
25
+
26
+ actor_critic:
27
+ _target_: models.actor_critic.ActorCriticConfig
28
+ lstm_dim: 512
29
+ img_channels: ${agent.denoiser.inner_model.img_channels}
30
+ img_size: ${env.train.size}
31
+ channels: [32,32,64,64]
32
+ down: [1,1,1,1]
33
+
atari_100k/config/env/atari.yaml ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ train:
2
+ id: null
3
+ done_on_life_loss: True
4
+ size: 64
5
+ max_episode_steps: null
6
+
7
+ test:
8
+ id: ${..train.id}
9
+ done_on_life_loss: False
10
+ size: ${..train.size}
11
+ max_episode_steps: null
12
+
13
+ keymap: atari/${.train.id}
atari_100k/models/Alien.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ae9cecd31f3126c18ed40fe0443f179388bd8295f65eb7ba0aaed35f805834eb
3
+ size 54348893
atari_100k/models/Amidar.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e2d53d508dbb7ba1ee34e12d6a0b2ae3bfac3e697d0b19a9968f4f3701f98321
3
+ size 54298054
atari_100k/models/Assault.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aa75ff8ca50630f42eabf1357c2c410ea3b23680ded7fd0c365e76a581308206
3
+ size 54317853
atari_100k/models/Asterix.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:21eb2a37537d78f525950a5e1d8304e17e7df773df253910f4255e7fac91d366
3
+ size 54323485
atari_100k/models/BankHeist.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8bf58c330b1a2b4257ff080fa48cb1093c436f3a948cc412401436f6021fc416
3
+ size 54348893
atari_100k/models/BattleZone.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6b70aaecd64b8fe4beb74b78799f498762d8bd47597da72fb13b17862c543910
3
+ size 54348893
atari_100k/models/Boxing.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fcb54ee18a7a19371800c86f18afcf78050bd453df99878601d3be9e0e7a5e07
3
+ size 54348893
atari_100k/models/Breakout.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:08404d2c49d23c49c211243b11af1a0d57d467bf3b25b94bcca9eb3aef1c63db
3
+ size 54309405
atari_100k/models/ChopperCommand.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f47785b7c746baa041a136cd49786766256a1b492b94e7745fa24fce40ccf4b5
3
+ size 54348893
atari_100k/models/CrazyClimber.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f525cfd71f080cdcc32c7070f9dfbec7b106b8c80d9cc84bbf68c768e9c122cf
3
+ size 54323485
atari_100k/models/DemonAttack.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8d62399c9e3b80a406cd27867a467e82c7019a2c7b016c947e123404092a1800
3
+ size 54315037
atari_100k/models/Freeway.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fdc18d43fa8ee5c66e62ca3544d34e42d0395b4bb7b5e6792faca9859cdc029e
3
+ size 54306589
atari_100k/models/Frostbite.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3592e044631a03fff3e36cb5e6c12202ab75bb86c98027720963604c161d3d64
3
+ size 54348893
atari_100k/models/Gopher.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8cc9c43e47e2ba05398df15594fc41eeaf4369c4667aed4759fed4539fbb5546
3
+ size 54320669
atari_100k/models/Hero.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c7fd037a423d13a1cfb53f2811f60c413ba7772960121704423c198799638538
3
+ size 54348893
atari_100k/models/Jamesbond.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b5c266ab4fe46c223409adb9e77e97211e56ec08cdcba1c0958418919a7a2a08
3
+ size 54348893
atari_100k/models/Kangaroo.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f7aacbf2fb3e9b14c0e2038964f09157a093a170fb00f20bb17c9581dddc2e1e
3
+ size 54348893
atari_100k/models/Krull.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f82685071e0595ce73dcc80ecdc642a849a5050f01d2eaa653320315c66d3d38
3
+ size 54348893
atari_100k/models/KungFuMaster.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ca9d8c0979344d7e91c9b299d0bde42db904f0ccde72ed3c507f5ed5fcd69444
3
+ size 54337629
atari_100k/models/MsPacman.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:91b213ef3f9e909130eb701424c44e4b92c53556974a63e571e23fec7f8028bd
3
+ size 54323485
atari_100k/models/Pong.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:addd34bd91f8c8e9c5c7f954c873d4db9fb42440c9028e93bf5e434c5ecb9042
3
+ size 54315037
atari_100k/models/PrivateEye.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4c92ed00ff97f97883b59b7d9961221d8e901cb3fc8ab3248a7c9ec92b318310
3
+ size 54348893
atari_100k/models/Qbert.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d13f6003b121d7e9300e95d5b0beb148a92541cf2a18cec8c69314b329f22f42
3
+ size 54315037
atari_100k/models/RoadRunner.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c410f0ec1a8f5f696041907443bbb74d80f2ad11f4ccf954bcf348fbc18f5ca0
3
+ size 54348893
atari_100k/models/Seaquest.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5a3413a60785c8f3753d56819b39a4f903af765300fae8a1b6c179f64349cc0b
3
+ size 54348893
atari_100k/models/UpNDown.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b6027efdba16f14ae732a2f2672f47d0668452cce0a183a443843f9061672e7c
3
+ size 54315037