autoprogrammer commited on
Commit
24e35c2
·
verified ·
1 Parent(s): 9384b5d

Update generation_utils.py

Browse files
Files changed (1) hide show
  1. generation_utils.py +1 -1
generation_utils.py CHANGED
@@ -171,7 +171,7 @@ class DreamGenerationMixin:
171
  total_steps: int,
172
  s: torch.Tensor,
173
  t: torch.Tensor,
174
- ema_beta: float = 0.8 # EMA 平滑系数(越大越稳定)
175
  ):
176
  """
177
  策略要点(接近 vanilla):
 
171
  total_steps: int,
172
  s: torch.Tensor,
173
  t: torch.Tensor,
174
+ ema_beta: float = 0.95 # EMA 平滑系数(越大越稳定)
175
  ):
176
  """
177
  策略要点(接近 vanilla):