trainable_param_report.json

{
  "dit_core_only_unexpected_trainable_params": 0,
  "groups": {
    "_groot_model.action_head.model": {
      "total": 550386688,
      "trainable": 550386688
    },
    "_groot_model.action_head.vl_self_attention": {
      "total": 201433088,
      "trainable": 201433088
    },
    "_groot_model.action_head.vlln": {
      "total": 4096,
      "trainable": 4096
    },
    "sequence_mgd_head": {
      "total": 0,
      "trainable": 0
    }
  },
  "num_learnable_params": 1068806144,
  "num_total_params": 2724163520,
  "other_trainable_params": 1068806144,
  "policy_type": "groot",
  "selected_prefixes": [],
  "trainable_mode": null,
  "trainable_ratio": 0.3923428737493702
}