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
|
||||
v=tf.constant(1.0,dtype=tf.float32)
|
||||
|
||||
if splits>1:
|
||||
q,v=SplitLayer(splits)([q,v])
|
||||
#if splits>1:
|
||||
q,v=SplitLayer(splits)([q,v])
|
||||
q,v=MotioLayer()([q,v])
|
||||
q,v=ScaleLayer()([q,v])
|
||||
for j in range(mixture):
|
||||
|
|
Loading…
Reference in New Issue