fixed bug

This commit is contained in:
Simon Klüttermann 2022-11-22 17:21:14 +01:00
parent c2fe311e6b
commit d6742f4198
1 changed files with 2 additions and 2 deletions

View File

@ -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):