frainer/task1_goalconditioned_envstate

1{2    "type": "act",3    "n_obs_steps": 1,4    "input_features": {5        "observation.state": {6            "type": "STATE",7            "shape": [8                69            ]10        },11        "observation.images.front": {12            "type": "VISUAL",13            "shape": [14                3,15                480,16                64017            ]18        },19        "observation.environment_state": {20            "type": "ENV",21            "shape": [22                323            ]24        }25    },26    "output_features": {27        "action": {28            "type": "ACTION",29            "shape": [30                631            ]32        }33    },34    "device": "cuda",35    "use_amp": false,36    "use_peft": false,37    "push_to_hub": true,38    "repo_id": "frainer/task1_goalconditioned_envstate",39    "private": null,40    "tags": null,41    "license": null,42    "pretrained_path": null,43    "chunk_size": 100,44    "n_action_steps": 100,45    "normalization_mapping": {46        "VISUAL": "MEAN_STD",47        "STATE": "MEAN_STD",48        "ACTION": "MEAN_STD"49    },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}