weights centroid Attention-UNet
weights based TensorFlow implementation for dropout manifold.
- Input
- 4523-dim embedding
- Encoder
- 95 x Attention-UNet with 2 heads
- Output
- f1 projection
Training config
optimizer=Adam, lr=0.634, scheduler=exponential, warmup=527