Skip to content

Commit

Permalink
Fix informer flake8
Browse files Browse the repository at this point in the history
  • Loading branch information
LongxingTan committed Jun 14, 2023
1 parent 6480fc6 commit 75f6d07
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tfts/models/informer.py
Original file line number Diff line number Diff line change
Expand Up @@ -122,7 +122,7 @@ def __call__(self, inputs, teacher=None):
B, L, _ = tf.shape(decoder_feature)
casual_mask = CausalMask(B * self.params["num_heads"], L).mask
decoder_feature = self.decoder_embedding(decoder_feature)

outputs = self.decoder(decoder_feature, memory=memory, x_mask=casual_mask)
outputs = self.projection(outputs)

Expand Down

0 comments on commit 75f6d07

Please sign in to comment.