File tree 1 file changed +0
-6
lines changed
1 file changed +0
-6
lines changed Original file line number Diff line number Diff line change @@ -242,7 +242,6 @@ def __init__(
242
242
self ,
243
243
domain_mods : Mapping [str , DomainModule ],
244
244
workspace_dim : int ,
245
- fusion_activation_fn : Callable [[torch .Tensor ], torch .Tensor ] = torch .tanh ,
246
245
* args ,
247
246
** kwargs ,
248
247
) -> None :
@@ -252,8 +251,6 @@ def __init__(
252
251
Args:
253
252
domain_modules (`Mapping[str, DomainModule]`): the domain modules.
254
253
workspace_dim (`int`): dimension of the GW.
255
- fusion_activation_fn (`Callable[[torch.Tensor], torch.Tensor]`): activation
256
- function used to fuse the domains.
257
254
"""
258
255
super ().__init__ ()
259
256
@@ -263,9 +260,6 @@ def __init__(
263
260
self .workspace_dim = workspace_dim
264
261
"""Dimension of the GW"""
265
262
266
- self .fusion_activation_fn = fusion_activation_fn
267
- """Activation function used to fuse the domains"""
268
-
269
263
@abstractmethod
270
264
def fuse (
271
265
self , x : LatentsDomainGroupT , selection_scores : Mapping [str , torch .Tensor ]
You can’t perform that action at this time.
0 commit comments