fixed bug
This commit is contained in:
parent
c2fe311e6b
commit
d6742f4198
4
model.py
4
model.py
|
@ -16,8 +16,8 @@ def gen_model(inputs, splits=10, realisation="gauss", mixture=0, nonlin=False):
|
||||||
q=i
|
q=i
|
||||||
v=tf.constant(1.0,dtype=tf.float32)
|
v=tf.constant(1.0,dtype=tf.float32)
|
||||||
|
|
||||||
if splits>1:
|
#if splits>1:
|
||||||
q,v=SplitLayer(splits)([q,v])
|
q,v=SplitLayer(splits)([q,v])
|
||||||
q,v=MotioLayer()([q,v])
|
q,v=MotioLayer()([q,v])
|
||||||
q,v=ScaleLayer()([q,v])
|
q,v=ScaleLayer()([q,v])
|
||||||
for j in range(mixture):
|
for j in range(mixture):
|
||||||
|
|
Loading…
Reference in New Issue