Yuchan
commited on
Update AlphaS2S.py
Browse files- 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):
|