notwa/optim
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

fix shape assertions

Browse source
This commit is contained in:
Connor Olding 2017-04-10 10:34:58 +00:00
parent a78fc98215
commit 0619163447
2 changed files with 3 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
optim_nn.py
View file

@ -106,8 +106,7 @@ class LayerNorm(Layer):
shape = parent.output_shape
self.input_shape = shape
self.output_shape = shape
if len(shape) != 1:
return False
assert len(shape) == 1, shape
if self.affine:
self.gamma.shape = (shape[0],)
self.beta.shape = (shape[0],)
@ -159,8 +158,7 @@ class Denses(Layer): # TODO: rename?
def make_shape(self, parent):
shape = parent.output_shape
self.input_shape = shape
if len(shape) != 2:
return False
assert len(shape) == 2, shape
assert -len(shape) <= self.axis < len(shape)
self.axis = self.axis % len(shape)

3
optim_nn_core.py
View file

@ -585,8 +585,7 @@ class Dense(Layer):
def make_shape(self, parent):
shape = parent.output_shape
self.input_shape = shape
if len(shape) != 1:
return False
assert len(shape) == 1, shape
self.coeffs.shape = (shape[0], self.dim)
self.biases.shape = (1, self.dim)