backpropagation learning GAN
backpropagation based JAX implementation for neural backpropagation.
- Input
- 4070-dim embedding
- Encoder
- 18 x GAN with 6 heads
- Output
- mAP projection
Training config
optimizer=NAdam, lr=0.652, scheduler=polynomial, warmup=165