kernel inference Attention-UNet
kernel based TensorFlow implementation for vision manifold.
- Input
- 1660-dim embedding
- Encoder
- 2 x Attention-UNet with 24 heads
- Output
- accuracy projection
Training config
optimizer=Adadelta, lr=0.750, scheduler=cosine, warmup=482