lychiyu/freqtrade_origin
1
0
Fork 0
mirror of https://github.com/freqtrade/freqtrade.git synced 2024-11-10 10:21:59 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

prevent mypy error, explicitly unpack input list of pytorch mlp model,

Browse Source
This commit is contained in:
Yinon Polak 2023-04-03 18:10:47 +03:00
parent d9d9993179
commit 0c4574b3b7
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
freqtrade/freqai/torch/PyTorchMLPModel.py
View File

@ -48,7 +48,7 @@ class PyTorchMLPModel(nn.Module):
self.dropout = nn.Dropout(p=dropout_percent)
def forward(self, x: List[torch.Tensor]) -> torch.Tensor:
x, = x
x = x[0]
x = self.relu(self.input_layer(x))
x = self.dropout(x)
x = self.blocks(x)