def update_ema()

in train_helpers.py [0:0]

• 4 lines of code
• 2 McCabe index (conditional complexity)

def update_ema(vae, ema_vae, ema_rate):
    for p1, p2 in zip(vae.parameters(), ema_vae.parameters()):
        p2.data.mul_(ema_rate)
        p2.data.add_(p1.data * (1 - ema_rate))