@@ -61,14 +61,13 @@ def __init__(
6161 timestep = input_shape [1 ]
6262 x = tf .unstack (self .inputs , axis = 1 )
6363
64- with utils .suppress_stdout ():
65- for i in range (0 , timestep ):
66- with tf .variable_scope (name , reuse = (set_keep ['name_reuse' ] if i == 0 else True )) as vs :
67- set_name_reuse ((set_keep ['name_reuse' ] if i == 0 else True ))
68- net = layer_class (InputLayer (x [i ], name = args ['name' ] + str (i )), ** args )
69- # net = layer_class(InputLayer(x[i], name="input_"+args['name']), **args)
70- x [i ] = net .outputs
71- variables = tf .get_collection (TF_GRAPHKEYS_VARIABLES , scope = vs .name )
64+ for i in range (0 , timestep ):
65+ with tf .variable_scope (name , reuse = (set_keep ['name_reuse' ] if i == 0 else True )) as vs :
66+ set_name_reuse ((set_keep ['name_reuse' ] if i == 0 else True ))
67+ net = layer_class (InputLayer (x [i ], name = args ['name' ] + str (i )), ** args )
68+ # net = layer_class(InputLayer(x[i], name="input_"+args['name']), **args)
69+ x [i ] = net .outputs
70+ variables = tf .get_collection (TF_GRAPHKEYS_VARIABLES , scope = vs .name )
7271
7372 self .outputs = tf .stack (x , axis = 1 , name = name )
7473
0 commit comments