Yuchan commited on
Commit
706457f
ยท
verified ยท
1 Parent(s): 957057b

Update AlphaS2S.py

Browse files
Files changed (1) hide show
  1. AlphaS2S.py +2 -0
AlphaS2S.py CHANGED
@@ -238,6 +238,8 @@ class gMLPBlock(layers.Layer):
238
  class CrossBlock(layers.Layer):
239
  def __init__(self, clip_value=5.0, eps=1e-6): # ๐Ÿ’ก d_model ์ธ์ž ์ถ”๊ฐ€
240
  super().__init__()
 
 
241
  # ๐Ÿ’ก ์ˆ˜์ •: ์ถœ๋ ฅ ์ฐจ์›์„ 1์—์„œ d_model๋กœ ๋ณ€๊ฒฝ
242
  def call(self, x, z):
243
  # a์˜ shape: (Batch, Seq_len, D_model)
 
238
  class CrossBlock(layers.Layer):
239
  def __init__(self, clip_value=5.0, eps=1e-6): # ๐Ÿ’ก d_model ์ธ์ž ์ถ”๊ฐ€
240
  super().__init__()
241
+ self.clip_value = clip_value
242
+ self.eps = eps
243
  # ๐Ÿ’ก ์ˆ˜์ •: ์ถœ๋ ฅ ์ฐจ์›์„ 1์—์„œ d_model๋กœ ๋ณ€๊ฒฝ
244
  def call(self, x, z):
245
  # a์˜ shape: (Batch, Seq_len, D_model)