Skip to content

Commit 863cbd7

Browse files
committed
Update layers.py
1 parent cf024f4 commit 863cbd7

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

torch_rechub/basic/layers.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -918,6 +918,8 @@ def forward(self, x, rel_pos_bias=None):
918918
for layer in self.layers:
919919
x = layer(x, rel_pos_bias=rel_pos_bias)
920920
return x
921+
922+
921923
class InteractingLayer(nn.Module):
922924
"""Multi-head Self-Attention based Interacting Layer, used in AutoInt model.
923925

0 commit comments

Comments
 (0)