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") d_b=Gradient("b")
) )
class PrecomputableMaxouts(Model): 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) Model.__init__(self, **kwargs)
self.nO = nO self.nO = nO
self.nP = pieces self.nP = nP
self.nI = nI self.nI = nI
self.nF = nF self.nF = nF