Skip to content

Commit 69aae58

Browse files
committed
Fix get_config() of PositionalEncoding
1 parent 37ea12a commit 69aae58

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tensorflow_asr/models/layers/positional_encoding.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,8 @@ def call(
6868

6969
def get_config(self):
7070
conf = super().get_config()
71-
return conf.update({"alpha": self.alpha, "beta": self.beta})
71+
conf.update({"alpha": self.alpha, "beta": self.beta})
72+
return conf
7273

7374

7475
class PositionalEncodingConcat(PositionalEncoding):

0 commit comments

Comments
 (0)