qiuhui1991

def update_ema_variables(model, ema_model, global_step): alpha = min(1 - 1 / (global_step + 1), 0.999) for ema_param, param in zip(ema_model.parameters(), model.parameters()): ema_param.data.mul_(alpha).add_(1 - alpha, param.data) in different model, the function update_ema_variables...