Update config.json
Browse files- config.json +0 -11
config.json
CHANGED
|
@@ -49,24 +49,13 @@
|
|
| 49 |
"ACTION": "MIN_MAX"
|
| 50 |
},
|
| 51 |
"vision_backbone": "resnet18",
|
| 52 |
-
"crop_shape": [
|
| 53 |
-
84,
|
| 54 |
-
84
|
| 55 |
-
],
|
| 56 |
-
"crop_is_random": true,
|
| 57 |
"pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
|
| 58 |
-
"use_group_norm": true,
|
| 59 |
-
"spatial_softmax_num_keypoints": 32,
|
| 60 |
-
"use_separate_rgb_encoder_per_camera": false,
|
| 61 |
"state_dim": 7,
|
| 62 |
"action_dim": 7,
|
| 63 |
"d_model": 128,
|
| 64 |
"nhead": 4,
|
| 65 |
"num_encoder_layers": 4,
|
| 66 |
-
"num_decoder_layers": 4,
|
| 67 |
"dim_feedforward": 512,
|
| 68 |
-
"dropout": 0.1,
|
| 69 |
-
"activation": "relu",
|
| 70 |
"state_token_dim": 64,
|
| 71 |
"optimizer_lr": 0.0001,
|
| 72 |
"optimizer_betas": [
|
|
|
|
| 49 |
"ACTION": "MIN_MAX"
|
| 50 |
},
|
| 51 |
"vision_backbone": "resnet18",
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 52 |
"pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
|
|
|
|
|
|
|
|
|
|
| 53 |
"state_dim": 7,
|
| 54 |
"action_dim": 7,
|
| 55 |
"d_model": 128,
|
| 56 |
"nhead": 4,
|
| 57 |
"num_encoder_layers": 4,
|
|
|
|
| 58 |
"dim_feedforward": 512,
|
|
|
|
|
|
|
| 59 |
"state_token_dim": 64,
|
| 60 |
"optimizer_lr": 0.0001,
|
| 61 |
"optimizer_betas": [
|