pravsels commited on
Commit
45b4cf8
·
verified ·
1 Parent(s): f28b1af

Upload policy weights, train config and readme

Browse files
Files changed (3) hide show
  1. README.md +2 -2
  2. config.json +3 -3
  3. train_config.json +3 -3
README.md CHANGED
@@ -5,9 +5,9 @@ license: apache-2.0
5
  model_name: rewact
6
  pipeline_tag: robotics
7
  tags:
8
- - robotics
9
- - lerobot
10
  - rewact
 
 
11
  ---
12
 
13
  # Model Card for rewact
 
5
  model_name: rewact
6
  pipeline_tag: robotics
7
  tags:
 
 
8
  - rewact
9
+ - lerobot
10
+ - robotics
11
  ---
12
 
13
  # Model Card for rewact
config.json CHANGED
@@ -38,7 +38,7 @@
38
  ]
39
  }
40
  },
41
- "device": "cuda",
42
  "use_amp": false,
43
  "push_to_hub": true,
44
  "repo_id": "pravsels/so100_rewact_dinov3",
@@ -47,8 +47,8 @@
47
  "license": null,
48
  "chunk_size": 100,
49
  "n_action_steps": 100,
50
- "vision_backbone": "resnet18",
51
- "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
52
  "replace_final_stride_with_dilation": false,
53
  "pre_norm": false,
54
  "dim_model": 512,
 
38
  ]
39
  }
40
  },
41
+ "device": null,
42
  "use_amp": false,
43
  "push_to_hub": true,
44
  "repo_id": "pravsels/so100_rewact_dinov3",
 
47
  "license": null,
48
  "chunk_size": 100,
49
  "n_action_steps": 100,
50
+ "vision_backbone": "dinov3_vitb16",
51
+ "pretrained_backbone_weights": "./dinov3_vitb16_pretrain_lvd1689m-73cec8be.pth",
52
  "replace_final_stride_with_dilation": false,
53
  "pre_norm": false,
54
  "dim_model": 512,
train_config.json CHANGED
@@ -105,7 +105,7 @@
105
  ]
106
  }
107
  },
108
- "device": "cuda",
109
  "use_amp": false,
110
  "push_to_hub": true,
111
  "repo_id": "pravsels/so100_rewact_dinov3",
@@ -114,8 +114,8 @@
114
  "license": null,
115
  "chunk_size": 100,
116
  "n_action_steps": 100,
117
- "vision_backbone": "resnet18",
118
- "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
119
  "replace_final_stride_with_dilation": false,
120
  "pre_norm": false,
121
  "dim_model": 512,
 
105
  ]
106
  }
107
  },
108
+ "device": null,
109
  "use_amp": false,
110
  "push_to_hub": true,
111
  "repo_id": "pravsels/so100_rewact_dinov3",
 
114
  "license": null,
115
  "chunk_size": 100,
116
  "n_action_steps": 100,
117
+ "vision_backbone": "dinov3_vitb16",
118
+ "pretrained_backbone_weights": "./dinov3_vitb16_pretrain_lvd1689m-73cec8be.pth",
119
  "replace_final_stride_with_dilation": false,
120
  "pre_norm": false,
121
  "dim_model": 512,