Upload policy weights, train config and readme
Browse files- README.md +1 -1
- config.json +1 -1
- train_config.json +2 -2
README.md
CHANGED
|
@@ -5,8 +5,8 @@ license: apache-2.0
|
|
| 5 |
model_name: act
|
| 6 |
pipeline_tag: robotics
|
| 7 |
tags:
|
| 8 |
-
- robotics
|
| 9 |
- act
|
|
|
|
| 10 |
- lerobot
|
| 11 |
---
|
| 12 |
|
|
|
|
| 5 |
model_name: act
|
| 6 |
pipeline_tag: robotics
|
| 7 |
tags:
|
|
|
|
| 8 |
- act
|
| 9 |
+
- robotics
|
| 10 |
- lerobot
|
| 11 |
---
|
| 12 |
|
config.json
CHANGED
|
@@ -35,7 +35,7 @@
|
|
| 35 |
"n_action_steps": 100,
|
| 36 |
"vision_backbone": "resnet18",
|
| 37 |
"pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
|
| 38 |
-
"replace_final_stride_with_dilation":
|
| 39 |
"pre_norm": false,
|
| 40 |
"dim_model": 512,
|
| 41 |
"n_heads": 8,
|
|
|
|
| 35 |
"n_action_steps": 100,
|
| 36 |
"vision_backbone": "resnet18",
|
| 37 |
"pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
|
| 38 |
+
"replace_final_stride_with_dilation": 0,
|
| 39 |
"pre_norm": false,
|
| 40 |
"dim_model": 512,
|
| 41 |
"n_heads": 8,
|
train_config.json
CHANGED
|
@@ -102,7 +102,7 @@
|
|
| 102 |
"n_action_steps": 100,
|
| 103 |
"vision_backbone": "resnet18",
|
| 104 |
"pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
|
| 105 |
-
"replace_final_stride_with_dilation":
|
| 106 |
"pre_norm": false,
|
| 107 |
"dim_model": 512,
|
| 108 |
"n_heads": 8,
|
|
@@ -122,7 +122,7 @@
|
|
| 122 |
},
|
| 123 |
"output_dir": "sim_policy/pushT_bottom_right_act",
|
| 124 |
"job_name": "act",
|
| 125 |
-
"resume":
|
| 126 |
"seed": 100000,
|
| 127 |
"num_workers": 4,
|
| 128 |
"batch_size": 64,
|
|
|
|
| 102 |
"n_action_steps": 100,
|
| 103 |
"vision_backbone": "resnet18",
|
| 104 |
"pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
|
| 105 |
+
"replace_final_stride_with_dilation": 0,
|
| 106 |
"pre_norm": false,
|
| 107 |
"dim_model": 512,
|
| 108 |
"n_heads": 8,
|
|
|
|
| 122 |
},
|
| 123 |
"output_dir": "sim_policy/pushT_bottom_right_act",
|
| 124 |
"job_name": "act",
|
| 125 |
+
"resume": true,
|
| 126 |
"seed": 100000,
|
| 127 |
"num_workers": 4,
|
| 128 |
"batch_size": 64,
|