backpropagation clustering Diffusion
backpropagation based PyTorch implementation for stable-diffusion environment.
- Input
- 7888-dim embedding
- Encoder
- 88 x Diffusion with 28 heads
- Output
- mAP projection
Training config
optimizer=Adam, lr=0.857, scheduler=exponential, warmup=806
