ISdept commited on
Commit
60accb5
·
verified ·
1 Parent(s): 87aa12d

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. config.json +5 -4
  2. 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": 256,
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
- "optimizer_lr": 0.000014,
 
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": 26000,
98
- "training_epoch": 68
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:0a48963883a064068158f9612ea7e195f20585c262dd7e6784719e6950b3aa5b
3
  size 745509700
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0f8d6debe5b4935053a05e313daa43c8dc7920177d0b3889218f24133b62f48d
3
  size 745509700