BYOL's target network is not trained by gradients. After every step it moves a small fraction toward the online network, which gives the target a slowly changing, stable identity.
Implement ema_update(target, online, tau), returning (1 - tau) * target + tau * online.
tau is small in practice, such as 0.004 or 0.01.Blend the two parameter arrays with weights 1 - tau and tau.
No results yet
Run your code to check it against the examples, or submit to run every test.