File size: 1,456 Bytes
b520d43
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
---
library_name: diffusers
license: other
license_name: research-use
tags:
  - object-detection
  - bounding-boxes
  - grounding
  - diffusion
  - synthetic-data
---

# socknethd

Higher-resolution 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 | 4,125,000 |
| Training run | [W&B](https://wandb.ai/diffusionexp/train_boxnethd/runs/c60hovpk) |

## Contents

| Path | Size | What it is |
|---|---|---|
| `boxnet/` | 16.1 GB | Joint image + bounding-box diffusion backbone |
| `pytorch_lora_weights.safetensors` | 2.8 GB | LoRA adapter weights |
| `nash_mtl_weights_conf_ema.pkl` | 432 B | Nash-MTL task-weighting coefficients |
| `nash_mtl_weights_ema.pkl` | 424 B | Nash-MTL task-weighting coefficients |

**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_socknethd,
  title  = {socknethd},
  author = {Wang, Franklin and Zhong, Desmond and Murdoch, Jamie},
  year   = {2026},
  url    = {https://huggingface.co/BreakpointAI/socknethd}
}
```