Trizz123/il_sim_test

1{2    "type": "act",3    "n_obs_steps": 1,4    "normalization_mapping": {5        "VISUAL": "MEAN_STD",6        "STATE": "MEAN_STD",7        "ACTION": "MEAN_STD"8    },9    "input_features": {10        "observation.state": {11            "type": "STATE",12            "shape": [13                1814            ]15        },16        "observation.images.front": {17            "type": "VISUAL",18            "shape": [19                3,20                128,21                12822            ]23        },24        "observation.images.wrist": {25            "type": "VISUAL",26            "shape": [27                3,28                128,29                12830            ]31        }32    },33    "output_features": {34        "action": {35            "type": "ACTION",36            "shape": [37                438            ]39        }40    },41    "device": "cuda",42    "use_amp": false,43    "push_to_hub": true,44    "repo_id": "Trizz123/il_sim_test",45    "private": null,46    "tags": null,47    "license": null,48    "chunk_size": 100,49    "n_action_steps": 100,50    "vision_backbone": "resnet18",51    "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",52    "replace_final_stride_with_dilation": false,53    "pre_norm": false,54    "dim_model": 512,55    "n_heads": 8,56    "dim_feedforward": 3200,57    "feedforward_activation": "relu",58    "n_encoder_layers": 4,59    "n_decoder_layers": 1,60    "use_vae": true,61    "latent_dim": 32,62    "n_vae_encoder_layers": 4,63    "temporal_ensemble_coeff": null,64    "dropout": 0.1,65    "kl_weight": 10.0,66    "optimizer_lr": 1e-05,67    "optimizer_weight_decay": 0.0001,68    "optimizer_lr_backbone": 1e-0569}