clustering reward Attention-UNet
clustering based vLLM implementation for cluster centroid.
- Input
- 1515-dim embedding
- Encoder
- 124 x Attention-UNet with 6 heads
- Output
- auc-roc projection
Training config
optimizer=Adagrad, lr=0.761, scheduler=exponential, warmup=1638
