From b4a120c745a7cbba907c5f1392b8b7052a3bb8b2 Mon Sep 17 00:00:00 2001 From: bubbliiiing <3323290568@qq.com> Date: Wed, 28 Oct 2026 10:59:32 +0800 Subject: [PATCH] Update ema ratio --- videox_fun/utils/fsdp_ema.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/videox_fun/utils/fsdp_ema.py b/videox_fun/utils/fsdp_ema.py index 8bf2ece..54977e8 100644 --- a/videox_fun/utils/fsdp_ema.py +++ b/videox_fun/utils/fsdp_ema.py @@ -37,7 +37,7 @@ class FSDPEMA: at the cost of one full fp32 copy in host RAM per rank). """ - def __init__(self, module, source, accelerator, fsdp_plugin, decay=0.9999): + def __init__(self, module, source, accelerator, fsdp_plugin, decay=0.99): from torch.distributed.fsdp import FullyShardedDataParallel as FSDP self.accelerator = accelerator