Function inplace_lerp

diff2flow/ema.py:26–30

(src: Tensor, tgt: Tensor, weight, *, auto_move_device = False)

    src.copy_(tgt)

def inplace_lerp(src: Tensor, tgt: Tensor, weight, *, auto_move_device = False):
    if auto_move_device:
        tgt = tgt.to(src.device)

    src.lerp_(tgt, weight)

class EMA(Module):
    """