Update config.json

config.json

@@ -33,7 +33,7 @@
     "device": "cuda",
     "use_amp": false,
     "chunk_size": 100,
-    "n_action_steps": 1,
+    "n_action_steps": 50,
     "vision_backbone": "resnet18",
     "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
     "replace_final_stride_with_dilation": false,
@@ -47,7 +47,7 @@
     "use_vae": true,
     "latent_dim": 32,
     "n_vae_encoder_layers": 4,
-    "temporal_ensemble_coeff": 0.5,
+    "temporal_ensemble_coeff": null,
     "dropout": 0.1,
     "kl_weight": 10.0,
     "optimizer_lr": 1e-05,