Fix pieces argument to PrecomputedMaxout

This commit is contained in:
Matthew Honnibal 2017-05-25 06:46:59 -05:00
parent e1cb5be0c7
commit b27c587800

View File

@ -86,10 +86,10 @@ class PrecomputableAffine(Model):
d_b=Gradient("b")
)
class PrecomputableMaxouts(Model):
def __init__(self, nO=None, nI=None, nF=None, pieces=3, **kwargs):
def __init__(self, nO=None, nI=None, nF=None, nP=3, **kwargs):
Model.__init__(self, **kwargs)
self.nO = nO
self.nP = pieces
self.nP = nP
self.nI = nI
self.nF = nF