Add config.json for download tracking

#3
by moojink - opened
Files changed (1) hide show
  1. config.json +67 -0
config.json ADDED
@@ -0,0 +1,67 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "cosmos-policy",
3
+ "architecture": "diffusion-transformer",
4
+ "base_model": "nvidia/Cosmos-Predict2-2B-Video2World",
5
+ "num_parameters": "2B",
6
+
7
+ "input_spec": {
8
+ "text": {
9
+ "type": "string",
10
+ "description": "Natural language task description"
11
+ },
12
+ "images": {
13
+ "format": "RGB",
14
+ "resolution": [224, 224],
15
+ "views": ["agentview", "eye_in_hand"]
16
+ },
17
+ "proprioception": {
18
+ "dim": 9,
19
+ "components": ["gripper_joints", "end_effector_position", "quaternion"]
20
+ }
21
+ },
22
+
23
+ "output_spec": {
24
+ "actions": {
25
+ "dim": 7,
26
+ "horizon": 16,
27
+ "components": ["end_effector_6dof", "gripper"]
28
+ },
29
+ "future_proprioception": {
30
+ "dim": 9
31
+ },
32
+ "future_images": {
33
+ "resolution": [224, 224]
34
+ },
35
+ "value": {
36
+ "dim": 1
37
+ }
38
+ },
39
+
40
+ "diffusion_config": {
41
+ "denoising_steps": 5,
42
+ "sigma_min": 4.0,
43
+ "sigma_max": 80.0,
44
+ "generation_mode": "parallel"
45
+ },
46
+
47
+ "training": {
48
+ "dataset": "LIBERO-Cosmos-Policy",
49
+ "gradient_steps": 40000,
50
+ "batch_size": 1920,
51
+ "hardware": "64x H100",
52
+ "action_chunk_size": 16
53
+ },
54
+
55
+ "benchmark_results": {
56
+ "libero_spatial": 0.981,
57
+ "libero_object": 1.0,
58
+ "libero_goal": 0.982,
59
+ "libero_long": 0.976,
60
+ "average": 0.985
61
+ },
62
+
63
+ "inference": {
64
+ "precision": "bf16",
65
+ "vram_gb": 6.8
66
+ }
67
+ }