mirror of
https://github.com/freqtrade/freqtrade.git
synced 2024-09-20 09:31:12 +00:00
fix arch
This commit is contained in:
parent
3bbb7e38ea
commit
36e1e58dad
|
@ -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)),
|
||||
|
|
Loading…
Reference in New Issue
Block a user