Update nn.py

fix a simple bug
2020-11-28 20:27:07 +08:00 · 2020-11-28 20:27:07 +08:00 · 260a052db9
parent f9bc36b1c7
commit 260a052db9
1 changed files with 1 additions and 1 deletions
--- a/python/jittor/nn.py
+++ b/python/jittor/nn.py
@ -885,7 +885,7 @@ class Sigmoid(Module):
    def execute(self, x) :
        return x.sigmoid()

-def softplus(x,bata=1,threshold=20):
+def softplus(x,beta=1,threshold=20):
    return 1 / beta * jt.log(1 + (beta * x).exp())

 def hardtanh(x,min_val=-1,max_val=1):