config.json

{
  "action_decoder_type": "mlp",
  "encoder_type": "theaiinstitute/theia-base-patch16-224-cddsv",
  "freeze_encoder": false,
  "goal_embed": false,
  "goal_input_dim": 2,
  "goal_output_dim": 64,
  "input_size": [
    3,
    224,
    224
  ],
  "num_actions_parameters": 2,
  "num_flatten_channels": 6,
  "num_prediction_steps": 10,
  "pretrained_encoder": true,
  "state_fusion_type": "mlp"
}