• Glenn Jocher's avatar
    Update EMA decay `tau` (#6769) · 0f819919
    Glenn Jocher 提交于
    * Update EMA
    
    * Update EMA
    
    * ratio invert
    
    * fix ratio invert
    
    * fix2 ratio invert
    
    * warmup iterations to 100
    
    * ema_k
    
    * implement tau
    
    * implement tau
    0f819919
torch_utils.py 13.5 KB