{
  "model_class": "ActionAutoencoder",
  "total_params": 11790537,
  "trainable_params": 9306553,
  "total_params_M": 11.7905,
  "trainable_params_M": 9.3066,
  "modules": {
    "encoder": {
      "total_params": 2483984,
      "trainable_params": 2483984,
      "total_params_M": 2.484,
      "trainable_params_M": 2.484
    },
    "decoder": {
      "total_params": 2219784,
      "trainable_params": 2219784,
      "total_params_M": 2.2198,
      "trainable_params_M": 2.2198
    },
    "dynamics_model": {
      "total_params": 3496976,
      "trainable_params": 3496976,
      "total_params_M": 3.497,
      "trainable_params_M": 3.497
    },
    "progress_model": {
      "total_params": 1105425,
      "trainable_params": 1105425,
      "total_params_M": 1.1054,
      "trainable_params_M": 1.1054
    },
    "target_encoder": {
      "total_params": 2483984,
      "trainable_params": 0,
      "total_params_M": 2.484,
      "trainable_params_M": 0.0
    }
  }
}