Add config.json for download tracking

#2
Files changed (1) hide show
  1. config.json +69 -0
config.json ADDED
@@ -0,0 +1,69 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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_left", "agentview_right", "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": 32,
27
+ "execution_horizon": 16,
28
+ "components": ["end_effector_6dof", "gripper"]
29
+ },
30
+ "future_proprioception": {
31
+ "dim": 9
32
+ },
33
+ "future_images": {
34
+ "resolution": [224, 224],
35
+ "views": 3
36
+ },
37
+ "value": {
38
+ "dim": 1
39
+ }
40
+ },
41
+
42
+ "diffusion_config": {
43
+ "denoising_steps": 5,
44
+ "sigma_min": 4.0,
45
+ "sigma_max": 80.0,
46
+ "generation_mode": "parallel"
47
+ },
48
+
49
+ "training": {
50
+ "dataset": "RoboCasa-Cosmos-Policy",
51
+ "gradient_steps": 45000,
52
+ "batch_size": 800,
53
+ "hardware": "32x H100",
54
+ "action_chunk_size": 32,
55
+ "num_tasks": 24,
56
+ "demos_per_task": 50
57
+ },
58
+
59
+ "benchmark_results": {
60
+ "robocasa_average": 0.671
61
+ },
62
+
63
+ "inference": {
64
+ "precision": "bf16",
65
+ "vram_gb": 8.9
66
+ },
67
+
68
+ "simulation_environment": "RoboCasa"
69
+ }