Hugging Face's logo

p3nGu1nZz
/
Tau

ONNX
Model card Files Community
p3nGu1nZz commited on
Commit
83bab82
•
1 Parent(s): 2f6b98d

completed D series model checkpoints.

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. results/tau_agent_D11_110M/Tau-D11-110M.onnx +3 -0
  2. results/tau_agent_D11_110M/TauAgent/TauAgent-10002300.onnx +3 -0
  3. results/tau_agent_D11_110M/TauAgent/TauAgent-10002300.pt +3 -0
  4. results/tau_agent_D11_110M/TauAgent/TauAgent-1999712.onnx +3 -0
  5. results/tau_agent_D11_110M/TauAgent/TauAgent-1999712.pt +3 -0
  6. results/tau_agent_D11_110M/TauAgent/TauAgent-2999867.onnx +3 -0
  7. results/tau_agent_D11_110M/TauAgent/TauAgent-2999867.pt +3 -0
  8. results/tau_agent_D11_110M/TauAgent/TauAgent-3999723.onnx +3 -0
  9. results/tau_agent_D11_110M/TauAgent/TauAgent-3999723.pt +3 -0
  10. results/tau_agent_D11_110M/TauAgent/TauAgent-4999878.onnx +3 -0
  11. results/tau_agent_D11_110M/TauAgent/TauAgent-4999878.pt +3 -0
  12. results/tau_agent_D11_110M/TauAgent/TauAgent-5999734.onnx +3 -0
  13. results/tau_agent_D11_110M/TauAgent/TauAgent-5999734.pt +3 -0
  14. results/tau_agent_D11_110M/TauAgent/TauAgent-6999889.onnx +3 -0
  15. results/tau_agent_D11_110M/TauAgent/TauAgent-6999889.pt +3 -0
  16. results/tau_agent_D11_110M/TauAgent/TauAgent-7999745.onnx +3 -0
  17. results/tau_agent_D11_110M/TauAgent/TauAgent-7999745.pt +3 -0
  18. results/tau_agent_D11_110M/TauAgent/TauAgent-8999900.onnx +3 -0
  19. results/tau_agent_D11_110M/TauAgent/TauAgent-8999900.pt +3 -0
  20. results/tau_agent_D11_110M/TauAgent/TauAgent-9999900.onnx +3 -0
  21. results/tau_agent_D11_110M/TauAgent/TauAgent-9999900.pt +3 -0
  22. results/tau_agent_D11_110M/TauAgent/checkpoint.pt +3 -0
  23. results/tau_agent_D11_110M/configuration.yaml +95 -0
  24. results/tau_agent_D12_120M/Tau-D12-120M.onnx +3 -0
  25. results/tau_agent_D12_120M/TauAgent/TauAgent-10015005.onnx +3 -0
  26. results/tau_agent_D12_120M/TauAgent/TauAgent-10015005.pt +3 -0
  27. results/tau_agent_D12_120M/TauAgent/TauAgent-1999712.onnx +3 -0
  28. results/tau_agent_D12_120M/TauAgent/TauAgent-1999712.pt +3 -0
  29. results/tau_agent_D12_120M/TauAgent/TauAgent-2999867.onnx +3 -0
  30. results/tau_agent_D12_120M/TauAgent/TauAgent-2999867.pt +3 -0
  31. results/tau_agent_D12_120M/TauAgent/TauAgent-3999723.onnx +3 -0
  32. results/tau_agent_D12_120M/TauAgent/TauAgent-3999723.pt +3 -0
  33. results/tau_agent_D12_120M/TauAgent/TauAgent-4999878.onnx +3 -0
  34. results/tau_agent_D12_120M/TauAgent/TauAgent-4999878.pt +3 -0
  35. results/tau_agent_D12_120M/TauAgent/TauAgent-5999734.onnx +3 -0
  36. results/tau_agent_D12_120M/TauAgent/TauAgent-5999734.pt +3 -0
  37. results/tau_agent_D12_120M/TauAgent/TauAgent-657800.onnx +3 -0
  38. results/tau_agent_D12_120M/TauAgent/TauAgent-657800.pt +3 -0
  39. results/tau_agent_D12_120M/TauAgent/TauAgent-6999889.onnx +3 -0
  40. results/tau_agent_D12_120M/TauAgent/TauAgent-6999889.pt +3 -0
  41. results/tau_agent_D12_120M/TauAgent/TauAgent-7999745.onnx +3 -0
  42. results/tau_agent_D12_120M/TauAgent/TauAgent-7999745.pt +3 -0
  43. results/tau_agent_D12_120M/TauAgent/TauAgent-8999900.onnx +3 -0
  44. results/tau_agent_D12_120M/TauAgent/TauAgent-8999900.pt +3 -0
  45. results/tau_agent_D12_120M/TauAgent/TauAgent-9999756.onnx +3 -0
  46. results/tau_agent_D12_120M/TauAgent/TauAgent-9999756.pt +3 -0
  47. results/tau_agent_D12_120M/TauAgent/checkpoint.pt +3 -0
  48. results/tau_agent_D12_120M/configuration.yaml +95 -0
  49. results/tau_agent_D13_130M/Tau-D13-130M.onnx +3 -0
  50. results/tau_agent_D13_130M/TauAgent/TauAgent-10030875.onnx +3 -0
results/tau_agent_D11_110M/Tau-D11-110M.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b93447a483234592c5c6d2f6e07ebb06808404eb7e515eed33b962a4e51b2e7c
3
+ size 2186395
results/tau_agent_D11_110M/TauAgent/TauAgent-10002300.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b93447a483234592c5c6d2f6e07ebb06808404eb7e515eed33b962a4e51b2e7c
3
+ size 2186395
results/tau_agent_D11_110M/TauAgent/TauAgent-10002300.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b5af120cf500b57847cc4b0c87c869cd3c0ea9cd82f7953a98aeaa49acfe4ba7
3
+ size 15534410
results/tau_agent_D11_110M/TauAgent/TauAgent-1999712.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bbaf9d29599e6a0ad3c52797d9ebe24fe094e4cf5bca45905f5732c5258821bb
3
+ size 2186395
results/tau_agent_D11_110M/TauAgent/TauAgent-1999712.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:51cd707dfb14da5a691d2d067be095ec47795b2e1394977f928adb5c39f98a43
3
+ size 15534256
results/tau_agent_D11_110M/TauAgent/TauAgent-2999867.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6e928e87c579da58f3acf43dc04527748440c0d6fe3e269ba0315f6cd010d8a8
3
+ size 2186395
results/tau_agent_D11_110M/TauAgent/TauAgent-2999867.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2b24a7c85d1ba619c7e632c966480ba26c102f2423f182c80289a466394671e6
3
+ size 15534256
results/tau_agent_D11_110M/TauAgent/TauAgent-3999723.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:285ab7ad09379ae7e0b21fd96855335f5f4d9fdd5d1891869b1139492be98d5f
3
+ size 2186395
results/tau_agent_D11_110M/TauAgent/TauAgent-3999723.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:41d651a581239a502cab88d4bb3a124f7b8c0f4e058e9656de42cdd4e9374e37
3
+ size 15534256
results/tau_agent_D11_110M/TauAgent/TauAgent-4999878.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:af5aabc2cbd8d111d2915f9b548fe372cb2cb0cd93668a344bad88babefdc3d5
3
+ size 2186395
results/tau_agent_D11_110M/TauAgent/TauAgent-4999878.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f93d188769652ecaf6ddb5227325bb051365c4ee4b9f90779b148f5acbfbd406
3
+ size 15534256
results/tau_agent_D11_110M/TauAgent/TauAgent-5999734.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:813079a68d2bcac62e02e72eb3b07aa763683a2fa23e877bb1411b6d846275d7
3
+ size 2186395
results/tau_agent_D11_110M/TauAgent/TauAgent-5999734.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ae7ed6bd0234f292efe98c9043fe846f48c5534b2287588efd6079b04f1b52c2
3
+ size 15534256
results/tau_agent_D11_110M/TauAgent/TauAgent-6999889.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2a3e383c2594092bbbe85af9dd7e59978f2ad78b7709dc85746fef1e1eb876c6
3
+ size 2186395
results/tau_agent_D11_110M/TauAgent/TauAgent-6999889.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3f8666e681aa6946cfb92385c7d79016d44aec7401c95da85fd0e628db106de3
3
+ size 15534256
results/tau_agent_D11_110M/TauAgent/TauAgent-7999745.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4e8a8fd7a1189818dd6b341fc6c5e0cbf6849244b0c0e7b965cab3b6a858ddc2
3
+ size 2186395
results/tau_agent_D11_110M/TauAgent/TauAgent-7999745.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4175b42c47aa01a2039706c8dca7eccbb5e398d262a223ce6162507e337afdbc
3
+ size 15534256
results/tau_agent_D11_110M/TauAgent/TauAgent-8999900.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d5daf549a19d3d4812f6976e5dd3075650f4286b278f6ca614ff864715b5e70f
3
+ size 2186395
results/tau_agent_D11_110M/TauAgent/TauAgent-8999900.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ef8736eb4ab8c0bbaef28d38d67ebbd8669d72095b18aac5129b4ffb61d0204f
3
+ size 15534256
results/tau_agent_D11_110M/TauAgent/TauAgent-9999900.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:914160a68717d3a4964d0c8f29b039f7f50e86e738a90847439fb5c1abc3a059
3
+ size 2186395
results/tau_agent_D11_110M/TauAgent/TauAgent-9999900.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0d3d34c850f86eeb36355007d3048a0a0f05cfae9075b5a6c5704a23bb35d05e
3
+ size 15534256
results/tau_agent_D11_110M/TauAgent/checkpoint.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8031a423bd447f01590d6859f903ad169581ed6a7c81a34969d5778440ee58e0
3
+ size 15533332
results/tau_agent_D11_110M/configuration.yaml ADDED
@@ -0,0 +1,95 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ default_settings: null
2
+ behaviors:
3
+ TauAgent:
4
+ trainer_type: ppo
5
+ hyperparameters:
6
+ batch_size: 512
7
+ buffer_size: 4096
8
+ learning_rate: 3.0e-05
9
+ beta: 0.005
10
+ epsilon: 0.2
11
+ lambd: 0.95
12
+ num_epoch: 3
13
+ shared_critic: false
14
+ learning_rate_schedule: linear
15
+ beta_schedule: linear
16
+ epsilon_schedule: linear
17
+ checkpoint_interval: 1000000
18
+ network_settings:
19
+ normalize: true
20
+ hidden_units: 256
21
+ num_layers: 4
22
+ vis_encode_type: simple
23
+ memory:
24
+ sequence_length: 128
25
+ memory_size: 256
26
+ goal_conditioning_type: hyper
27
+ deterministic: false
28
+ reward_signals:
29
+ extrinsic:
30
+ gamma: 0.99
31
+ strength: 1.0
32
+ network_settings:
33
+ normalize: true
34
+ hidden_units: 256
35
+ num_layers: 4
36
+ vis_encode_type: simple
37
+ memory:
38
+ sequence_length: 128
39
+ memory_size: 256
40
+ goal_conditioning_type: hyper
41
+ deterministic: false
42
+ curiosity:
43
+ gamma: 0.995
44
+ strength: 0.1
45
+ network_settings:
46
+ normalize: true
47
+ hidden_units: 256
48
+ num_layers: 4
49
+ vis_encode_type: simple
50
+ memory: null
51
+ goal_conditioning_type: hyper
52
+ deterministic: false
53
+ learning_rate: 3.0e-05
54
+ encoding_size: null
55
+ init_path: results\tau_agent_D10\TauAgent\checkpoint.pt
56
+ keep_checkpoints: 10
57
+ even_checkpoints: true
58
+ max_steps: 10000000
59
+ time_horizon: 512
60
+ summary_freq: 100000
61
+ threaded: true
62
+ self_play: null
63
+ behavioral_cloning: null
64
+ env_settings:
65
+ env_path: .\Build
66
+ env_args: null
67
+ base_port: 5005
68
+ num_envs: 1
69
+ num_areas: 1
70
+ timeout_wait: 300
71
+ seed: -1
72
+ max_lifetime_restarts: 10
73
+ restarts_rate_limit_n: 1
74
+ restarts_rate_limit_period_s: 60
75
+ engine_settings:
76
+ width: 84
77
+ height: 84
78
+ quality_level: 5
79
+ time_scale: 20
80
+ target_frame_rate: -1
81
+ capture_frame_rate: 60
82
+ no_graphics: false
83
+ environment_parameters: null
84
+ checkpoint_settings:
85
+ run_id: tau_agent_D11
86
+ initialize_from: tau_agent_D10
87
+ load_model: false
88
+ resume: false
89
+ force: true
90
+ train_model: false
91
+ inference: false
92
+ results_dir: results
93
+ torch_settings:
94
+ device: cuda
95
+ debug: false
results/tau_agent_D12_120M/Tau-D12-120M.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2c9bbb49c2fb1448abfff70421ef63826da0400df1de1900c24ccde55ce57d13
3
+ size 2186395
results/tau_agent_D12_120M/TauAgent/TauAgent-10015005.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2c9bbb49c2fb1448abfff70421ef63826da0400df1de1900c24ccde55ce57d13
3
+ size 2186395
results/tau_agent_D12_120M/TauAgent/TauAgent-10015005.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:31488e308c5ece73d8118e73cdac38f7622044276bee3b28104f7b15d8a85d8b
3
+ size 15534410
results/tau_agent_D12_120M/TauAgent/TauAgent-1999712.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:334a037a38160272804cb79b9034385fedfd2346f7dadb5c20e8aa3a40ed716b
3
+ size 2186395
results/tau_agent_D12_120M/TauAgent/TauAgent-1999712.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8bc2fdba71d65d7e6bbe3d36bb73cd330166129db275cd00dfb419f2c82488b6
3
+ size 15534256
results/tau_agent_D12_120M/TauAgent/TauAgent-2999867.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:640f7fd8cefe5973f3e4eb4ce305ddaecf023f92eb85e53a842ed8fec1b91cfb
3
+ size 2186395
results/tau_agent_D12_120M/TauAgent/TauAgent-2999867.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c4a0f10b3160a3bebb7831d050c37d34d3115832b14e48e6f04464455970090f
3
+ size 15534256
results/tau_agent_D12_120M/TauAgent/TauAgent-3999723.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:306a63a4b24afd397cf2606decec6b54b8674846d19ab29ef7982f9edc8b5b73
3
+ size 2186395
results/tau_agent_D12_120M/TauAgent/TauAgent-3999723.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:465771084c0c7090124a5fbb21cea536fd7579ce49c4fef0d4f98da0af032c39
3
+ size 15534256
results/tau_agent_D12_120M/TauAgent/TauAgent-4999878.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f624483fd08603871fcf8c34e7fe94dbf07811bf52504f0353b3c8120d3b7e39
3
+ size 2186395
results/tau_agent_D12_120M/TauAgent/TauAgent-4999878.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9e45557347c222ee1e3374007e154096814b3db3f4cdb6c8aeb5039bc25d43bb
3
+ size 15534256
results/tau_agent_D12_120M/TauAgent/TauAgent-5999734.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0a3276271aec9c6e4731da7b4bbc55a9d96a5734d0cb27ca77a4a205a23a7731
3
+ size 2186395
results/tau_agent_D12_120M/TauAgent/TauAgent-5999734.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8a5508bd0cfd541bd472ca36297fb82a03de701ad212e10cb11b385bbe6e044c
3
+ size 15534256
results/tau_agent_D12_120M/TauAgent/TauAgent-657800.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9209eaec6d31cf6fb04e4c44350c0f8a96296580748e93f455f2784c7916531d
3
+ size 2186395
results/tau_agent_D12_120M/TauAgent/TauAgent-657800.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:52f44f14c760052cd3702e789f8278287e958eb94a58e53983b8c4f505e24349
3
+ size 15534102
results/tau_agent_D12_120M/TauAgent/TauAgent-6999889.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:40311cb454c6a053bfdafde3a9b87f89d1635e39e3574ccac5c651efb13ebe56
3
+ size 2186395
results/tau_agent_D12_120M/TauAgent/TauAgent-6999889.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:76f3f3364bcff7ce5a86533e3c27f2303d7ce94cd28c8539f6d18d8a8207f5ef
3
+ size 15534256
results/tau_agent_D12_120M/TauAgent/TauAgent-7999745.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:13acccd224287a412f1e7d0044ea66e84567202b332b50f51e12f17195ba65de
3
+ size 2186395
results/tau_agent_D12_120M/TauAgent/TauAgent-7999745.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d2bc9302228e9db24bae4318bc827c68907abccd83751a611a177975a0126b58
3
+ size 15534256
results/tau_agent_D12_120M/TauAgent/TauAgent-8999900.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e470a80060920e1229dcdca7ad54bead837a630b06bf36e6c85b7b388413439b
3
+ size 2186395
results/tau_agent_D12_120M/TauAgent/TauAgent-8999900.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bf4c652142bcb1c1afac8bc5d0f9c1af90ec63fe187ad4bfbbeb01732ed72722
3
+ size 15534256
results/tau_agent_D12_120M/TauAgent/TauAgent-9999756.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c656665d645c34b3d7a8e5df2f77f726300b674f4fe6c2d3144ea9471118e9c4
3
+ size 2186395
results/tau_agent_D12_120M/TauAgent/TauAgent-9999756.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:948794047d26d7d4e43bae12f3ea2bc161e41a8dc70c9fcd443446ffeff57427
3
+ size 15534256
results/tau_agent_D12_120M/TauAgent/checkpoint.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:167da69a09bfbaf0b2c24e6016140699791cc439c437f0f04fb586c1eda82207
3
+ size 15533332
results/tau_agent_D12_120M/configuration.yaml ADDED
@@ -0,0 +1,95 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ default_settings: null
2
+ behaviors:
3
+ TauAgent:
4
+ trainer_type: ppo
5
+ hyperparameters:
6
+ batch_size: 512
7
+ buffer_size: 4096
8
+ learning_rate: 3.0e-05
9
+ beta: 0.005
10
+ epsilon: 0.2
11
+ lambd: 0.95
12
+ num_epoch: 3
13
+ shared_critic: false
14
+ learning_rate_schedule: linear
15
+ beta_schedule: linear
16
+ epsilon_schedule: linear
17
+ checkpoint_interval: 1000000
18
+ network_settings:
19
+ normalize: true
20
+ hidden_units: 256
21
+ num_layers: 4
22
+ vis_encode_type: simple
23
+ memory:
24
+ sequence_length: 128
25
+ memory_size: 256
26
+ goal_conditioning_type: hyper
27
+ deterministic: false
28
+ reward_signals:
29
+ extrinsic:
30
+ gamma: 0.99
31
+ strength: 1.0
32
+ network_settings:
33
+ normalize: true
34
+ hidden_units: 256
35
+ num_layers: 4
36
+ vis_encode_type: simple
37
+ memory:
38
+ sequence_length: 128
39
+ memory_size: 256
40
+ goal_conditioning_type: hyper
41
+ deterministic: false
42
+ curiosity:
43
+ gamma: 0.995
44
+ strength: 0.1
45
+ network_settings:
46
+ normalize: true
47
+ hidden_units: 256
48
+ num_layers: 4
49
+ vis_encode_type: simple
50
+ memory: null
51
+ goal_conditioning_type: hyper
52
+ deterministic: false
53
+ learning_rate: 3.0e-05
54
+ encoding_size: null
55
+ init_path: results\tau_agent_D11\TauAgent\checkpoint.pt
56
+ keep_checkpoints: 10
57
+ even_checkpoints: true
58
+ max_steps: 10000000
59
+ time_horizon: 512
60
+ summary_freq: 100000
61
+ threaded: true
62
+ self_play: null
63
+ behavioral_cloning: null
64
+ env_settings:
65
+ env_path: .\Build
66
+ env_args: null
67
+ base_port: 5005
68
+ num_envs: 1
69
+ num_areas: 1
70
+ timeout_wait: 300
71
+ seed: -1
72
+ max_lifetime_restarts: 10
73
+ restarts_rate_limit_n: 1
74
+ restarts_rate_limit_period_s: 60
75
+ engine_settings:
76
+ width: 84
77
+ height: 84
78
+ quality_level: 5
79
+ time_scale: 20
80
+ target_frame_rate: -1
81
+ capture_frame_rate: 60
82
+ no_graphics: false
83
+ environment_parameters: null
84
+ checkpoint_settings:
85
+ run_id: tau_agent_D12
86
+ initialize_from: tau_agent_D11
87
+ load_model: false
88
+ resume: false
89
+ force: true
90
+ train_model: false
91
+ inference: false
92
+ results_dir: results
93
+ torch_settings:
94
+ device: cuda
95
+ debug: false
results/tau_agent_D13_130M/Tau-D13-130M.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f23c84a698d6fa0ce22f2fcdff4b7591e26e635aa7b82074034b3bc5eaf19e1b
3
+ size 2186395
results/tau_agent_D13_130M/TauAgent/TauAgent-10030875.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f23c84a698d6fa0ce22f2fcdff4b7591e26e635aa7b82074034b3bc5eaf19e1b
3
+ size 2186395