---
library_name: diffusers
license: other
license_name: research-use
tags:
  - object-detection
  - bounding-boxes
  - grounding
  - diffusion
  - synthetic-data
---

# socknetq

Joint image + bounding-box diffusion model with a LoRA adapter, a confidence stream, and Nash-MTL multi-task loss weighting.

Released by [Breakpoint AI](https://huggingface.co/BreakpointAI) as part of open-sourcing the company's research
artifacts.

| | |
|---|---|
| Training data | [`BreakpointAI/breakpoint-grounding-55m`](https://huggingface.co/datasets/BreakpointAI/breakpoint-grounding-55m) |
| Checkpoint step | 3,120,000 |
| Training run | [W&B](https://wandb.ai/diffusionexp/train_socknetq/runs/7pqundo1) |

## Contents

| Path | Size | What it is |
|---|---|---|
| `pytorch_model/` | 67.6 GB |  |
| `boxnet/` | 20.1 GB | Joint image + bounding-box diffusion backbone |
| `pytorch_lora_weights.safetensors` | 1.0 GB | LoRA adapter weights |
| `conf_weighter.pth` | 2.0 KB | Confidence-stream loss-weighting head |
| `weighter.pth` | 1.9 KB | Loss-weighting head |
| `nash_mtl_weights_conf_ema.safetensors` | 120 B | Nash-MTL task-weighting coefficients |
| `nash_mtl_weights_ema.safetensors` | 112 B | Nash-MTL task-weighting coefficients |
| `global_conf_step.txt` | 7 B | Training step counter |
| `global_nonull_step.txt` | 7 B | Training step counter |
| `weighter_steps.txt` | 4 B | Loss-weighter step counter |

**Inference weights only.** Optimizer, LR scheduler, RNG and dataloader state were not
uploaded, so this checkpoint cannot be used to resume training.

## Citation

```bibtex
@misc{breakpoint_socknetq,
  title  = {socknetq},
  author = {Wang, Franklin and Zhong, Desmond and Murdoch, Jamie},
  year   = {2026},
  url    = {https://huggingface.co/BreakpointAI/socknetq}
}
```