Update config.json
Browse files- config.json +6 -8
config.json
CHANGED
|
@@ -2,17 +2,15 @@
|
|
| 2 |
"architectures": [
|
| 3 |
"InterfuserForHuggingFace"
|
| 4 |
],
|
| 5 |
-
"
|
| 6 |
-
"dim_feedforward": 2048,
|
| 7 |
"embed_dim": 256,
|
| 8 |
"enc_depth": 6,
|
| 9 |
-
"
|
| 10 |
-
"lidar_backbone_name": "r18",
|
| 11 |
-
"model_type": "interfuser",
|
| 12 |
"num_heads": 8,
|
|
|
|
| 13 |
"rgb_backbone_name": "r50",
|
| 14 |
-
"
|
| 15 |
-
"
|
| 16 |
"use_different_backbone": true,
|
| 17 |
-
"
|
| 18 |
}
|
|
|
|
| 2 |
"architectures": [
|
| 3 |
"InterfuserForHuggingFace"
|
| 4 |
],
|
| 5 |
+
"model_type": "interfuser",
|
|
|
|
| 6 |
"embed_dim": 256,
|
| 7 |
"enc_depth": 6,
|
| 8 |
+
"dec_depth": 6,
|
|
|
|
|
|
|
| 9 |
"num_heads": 8,
|
| 10 |
+
"dim_feedforward": 2048,
|
| 11 |
"rgb_backbone_name": "r50",
|
| 12 |
+
"lidar_backbone_name": "r18",
|
| 13 |
+
"waypoints_pred_head": "gru",
|
| 14 |
"use_different_backbone": true,
|
| 15 |
+
"transformers_version": "4.41.0"
|
| 16 |
}
|