ensemble machine Attention-UNet
ensemble based JAX implementation for neural policy.
- Input
- 1520-dim embedding
- Encoder
- 86 x Attention-UNet with 14 heads
- Output
- accuracy projection
Training config
optimizer=RMSprop, lr=0.665, scheduler=cosine, warmup=1009