{
  "encoder": 5650176,
  "action_encoder": 39744,
  "predictor": 4037376,
  "total": 9801795,
  "ema_teacher_frozen": 5650176,
  "aux_heads": 74499
}