Fix init for padding

This commit is contained in:
Matthw Honnibal 2020-06-30 11:50:24 +02:00
parent 85ed5730a2
commit 233945bfe0

View File

@ -110,6 +110,7 @@ def init(model, X=None, Y=None):
ops = model.ops ops = model.ops
W = normal_init(ops, W.shape, mean=float(ops.xp.sqrt(1.0 / nF * nI))) W = normal_init(ops, W.shape, mean=float(ops.xp.sqrt(1.0 / nF * nI)))
pad = normal_init(ops, pad.shape, mean=1.0)
model.set_param("W", W) model.set_param("W", W)
model.set_param("b", b) model.set_param("b", b)
model.set_param("pad", pad) model.set_param("pad", pad)