mohammadian7/freqtrade
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix arch

Browse Source
This commit is contained in:
robcaulk
2023-05-06 17:40:04 +00:00
parent 3bbb7e38ea
commit 36e1e58dad
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
freqtrade/freqai/torch/PyTorchTransformerModel.py
+1 -1
View File
@@ -39,7 +39,7 @@ class PyTorchTransformerModel(nn.Module):
# the pseudo decoding FC # the pseudo decoding FC
self.output_net = nn.Sequential( 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.ReLU(),
nn.Dropout(dropout_percent), nn.Dropout(dropout_percent),
nn.Linear(int(hidden_dim), int(hidden_dim / 2)), nn.Linear(int(hidden_dim), int(hidden_dim / 2)),