Upload folder using huggingface_hub
Browse files- config.json +5 -4
- model.safetensors +1 -1
config.json
CHANGED
|
@@ -64,7 +64,7 @@
|
|
| 64 |
"ACTION": "MEAN_STD"
|
| 65 |
},
|
| 66 |
"freeze_vision_backbone": true,
|
| 67 |
-
"vision_input_size":
|
| 68 |
"num_cameras": 3,
|
| 69 |
"num_vlm_layers": 16,
|
| 70 |
"detection_classes": [
|
|
@@ -85,7 +85,8 @@
|
|
| 85 |
"dim_feedforward": 2048,
|
| 86 |
"diffusion_step_embed_dim": 128,
|
| 87 |
"num_inference_steps": 20,
|
| 88 |
-
"
|
|
|
|
| 89 |
"optimizer_betas": [
|
| 90 |
0.95,
|
| 91 |
0.999
|
|
@@ -94,6 +95,6 @@
|
|
| 94 |
"optimizer_weight_decay": 1e-06,
|
| 95 |
"scheduler_warmup_steps": 1500,
|
| 96 |
"use_peft": false,
|
| 97 |
-
"training_step":
|
| 98 |
-
"training_epoch":
|
| 99 |
}
|
|
|
|
| 64 |
"ACTION": "MEAN_STD"
|
| 65 |
},
|
| 66 |
"freeze_vision_backbone": true,
|
| 67 |
+
"vision_input_size": 384,
|
| 68 |
"num_cameras": 3,
|
| 69 |
"num_vlm_layers": 16,
|
| 70 |
"detection_classes": [
|
|
|
|
| 85 |
"dim_feedforward": 2048,
|
| 86 |
"diffusion_step_embed_dim": 128,
|
| 87 |
"num_inference_steps": 20,
|
| 88 |
+
"pos_decay_lambda": 0.1,
|
| 89 |
+
"optimizer_lr": 0.0001,
|
| 90 |
"optimizer_betas": [
|
| 91 |
0.95,
|
| 92 |
0.999
|
|
|
|
| 95 |
"optimizer_weight_decay": 1e-06,
|
| 96 |
"scheduler_warmup_steps": 1500,
|
| 97 |
"use_peft": false,
|
| 98 |
+
"training_step": 50000,
|
| 99 |
+
"training_epoch": 129
|
| 100 |
}
|
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 745509700
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0f8d6debe5b4935053a05e313daa43c8dc7920177d0b3889218f24133b62f48d
|
| 3 |
size 745509700
|