From 36e1e58dad0c63a41cf97792cea248942d590ca4 Mon Sep 17 00:00:00 2001 From: robcaulk Date: Sat, 6 May 2023 17:40:04 +0000 Subject: [PATCH] fix arch --- freqtrade/freqai/torch/PyTorchTransformerModel.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/freqtrade/freqai/torch/PyTorchTransformerModel.py b/freqtrade/freqai/torch/PyTorchTransformerModel.py index 2ab3ea434..702a7a08b 100644 --- a/freqtrade/freqai/torch/PyTorchTransformerModel.py +++ b/freqtrade/freqai/torch/PyTorchTransformerModel.py @@ -39,7 +39,7 @@ class PyTorchTransformerModel(nn.Module): # the pseudo decoding FC self.output_net = nn.Sequential( - nn.Linear(hidden_dim * time_window, int(hidden_dim)), + nn.Linear(self.dim_val * time_window, int(hidden_dim)), nn.ReLU(), nn.Dropout(dropout_percent), nn.Linear(int(hidden_dim), int(hidden_dim / 2)),