Skip to content

Commit 3828676

Browse files
committed
Make RmsNormAct sync with RmsNorm re default eps of 1e-6
1 parent 136440d commit 3828676

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

timm/layers/norm_act.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -478,7 +478,7 @@ class RmsNormAct(RmsNorm):
478478
def __init__(
479479
self,
480480
num_channels,
481-
eps=1e-5,
481+
eps=1e-6,
482482
affine=True,
483483
apply_act=True,
484484
act_layer=nn.ReLU,
@@ -511,7 +511,7 @@ class RmsNormAct2d(RmsNorm2d):
511511
def __init__(
512512
self,
513513
num_channels,
514-
eps=1e-5,
514+
eps=1e-6,
515515
affine=True,
516516
apply_act=True,
517517
act_layer=nn.ReLU,

0 commit comments

Comments
 (0)