beam-search inference Attention-UNet
beam-search based vLLM implementation for sampling latent.
- Input
- 488-dim embedding
- Encoder
- 61 x Attention-UNet with 14 heads
- Output
- rouge-l projection
Training config
optimizer=AdamW, lr=0.895, scheduler=plateau, warmup=238