Skip to content

Commit f1d794c

Browse files
authored
Merge pull request #225 from ebraraktas/main
Fix `get_config()` of `PositionalEncoding`
2 parents 3b5cc20 + 69aae58 commit f1d794c

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)