From a90b04283ceae04dacbc241cba2cf73e633804aa Mon Sep 17 00:00:00 2001 From: jgrss Date: Tue, 19 Dec 2023 18:54:39 +1100 Subject: [PATCH] correct filters --- src/cultionet/models/cultio.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/cultionet/models/cultio.py b/src/cultionet/models/cultio.py index b6cdcf73..9cd28bc6 100644 --- a/src/cultionet/models/cultio.py +++ b/src/cultionet/models/cultio.py @@ -317,11 +317,11 @@ def __init__( self.temporal_encoder = LightweightTemporalAttentionEncoder( in_channels=self.ds_num_bands, - hidden_size=filters, + hidden_size=self.filters, num_head=8, num_time=self.ds_num_time, dropout=0.1, - d_model=filters * 2, + d_model=self.filters, time_scaler=1_000, num_classes_l2=self.num_classes, num_classes_last=self.num_classes + 1,