From 1adfe4eb3564466e01bb5bdb87e91503a586b122 Mon Sep 17 00:00:00 2001 From: yehjin-shin Date: Fri, 18 Oct 2024 21:20:49 +0900 Subject: [PATCH] [FIX] correct default argument for layer-specific values --- src/model/fearec.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/model/fearec.py b/src/model/fearec.py index ca3ebde..e4a724b 100644 --- a/src/model/fearec.py +++ b/src/model/fearec.py @@ -170,7 +170,7 @@ def forward(self, hidden_states, attention_mask, output_all_encoded_layers=False class FEARecBlock(nn.Module): def __init__(self, args, layer_num): super(FEARecBlock, self).__init__() - self.layer = FEARecLayer(args) + self.layer = FEARecLayer(args, layer_num) self.feed_forward = FeedForward(args) def forward(self, hidden_states, attention_mask):