config.json

{
  "action_dim": 7,
  "architectures": [
    "ACTPolicy"
  ],
  "args": {
    "action_dim": 7,
    "backbone": "resnet18",
    "camera_names": [
      "primary",
      "wrist"
    ],
    "chunk_size": 16,
    "dec_layers": 7,
    "dilation": false,
    "dim_feedforward": 3200,
    "dropout": 0.1,
    "enc_layers": 4,
    "hidden_dim": 512,
    "history_len": 1,
    "image_size": [
      256,
      256
    ],
    "kl_weight": 10,
    "lr_backbone": 1e-05,
    "masks": false,
    "nheads": 8,
    "num_queries": 16,
    "position_embedding": "sine",
    "pre_norm": false,
    "prediction_len": 16,
    "raw_size": [
      640,
      480
    ],
    "state_dim": 7
  },
  "backbone": "resnet18",
  "camera_names": [
    "primary",
    "wrist"
  ],
  "chunk_size": 16,
  "dec_layers": 7,
  "dilation": false,
  "dim_feedforward": 3200,
  "dropout": 0.1,
  "enc_layers": 4,
  "hidden_dim": 512,
  "history_len": 1,
  "image_size": [
    256,
    256
  ],
  "kl_weight": 10,
  "lr_backbone": 1e-05,
  "masks": false,
  "nheads": 8,
  "num_queries": 16,
  "position_embedding": "sine",
  "pre_norm": false,
  "prediction_len": 16,
  "raw_size": [
    640,
    480
  ],
  "state_dim": 7,
  "torch_dtype": "float32",
  "transformers_version": "4.45.2",
  "type": "policy.act"
}