{
  "custom_modules": [
    "box_encoder",
    "lm_head",
    "model",
    "text_prompt_proj",
    "box_decoder"
  ],
  "total_params": 1504,
  "saved_at_step": 900
}