Skip to content

Commit beda9d7

Browse files
committed
The logit in #L44 set shape from itself, fix this with zero_logit
1 parent 67a4f4e commit beda9d7

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

seq_labeling.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ def copy_through():
4141

4242
logit = control_flow_ops.cond(
4343
time < min_sequence_length, existing_logit, copy_through)
44-
logit.set_shape(logit.get_shape())
44+
logit.set_shape(zero_logit.get_shape())
4545
return logit
4646

4747
def attention_RNN(encoder_outputs,
@@ -279,4 +279,4 @@ def generate_sequence_output(num_encoder_symbols,
279279
logits, bucket_target, weights,
280280
softmax_loss_function=softmax_loss_function)
281281

282-
return logits, crossent
282+
return logits, crossent

0 commit comments

Comments
 (0)