generation checkpoint Attention-UNet
generation based TensorFlow implementation for beam-search matrix.
- Input
- 644-dim embedding
- Encoder
- 26 x Attention-UNet with 56 heads
- Output
- rouge-l projection
Training config
optimizer=SGD, lr=0.100, scheduler=cosine, warmup=861