Yuchan commited on
Commit
957057b
·
verified ·
1 Parent(s): f566493

Update AlphaS2S.py

Browse files
Files changed (1) hide show
  1. AlphaS2S.py +1 -1
AlphaS2S.py CHANGED
@@ -236,7 +236,7 @@ class gMLPBlock(layers.Layer):
236
  return residual + out
237
 
238
  class CrossBlock(layers.Layer):
239
- def __init__(self): # 💡 d_model 인자 추가
240
  super().__init__()
241
  # 💡 수정: 출력 차원을 1에서 d_model로 변경
242
  def call(self, x, z):
 
236
  return residual + out
237
 
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):