From a87b215d67bce2fdd46455824f3ba63fdff0ab15 Mon Sep 17 00:00:00 2001 From: Matthias Date: Sun, 21 May 2023 09:50:59 +0200 Subject: [PATCH] Fix odd import --- freqtrade/freqai/torch/PyTorchTransformerModel.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/freqtrade/freqai/torch/PyTorchTransformerModel.py b/freqtrade/freqai/torch/PyTorchTransformerModel.py index 702a7a08b..162459776 100644 --- a/freqtrade/freqai/torch/PyTorchTransformerModel.py +++ b/freqtrade/freqai/torch/PyTorchTransformerModel.py @@ -1,7 +1,7 @@ import math import torch -import torch.nn as nn +from torch import nn """ @@ -68,7 +68,7 @@ class PyTorchTransformerModel(nn.Module): return x -class PositionalEncoding(torch.nn.Module): +class PositionalEncoding(nn.Module): def __init__(self, d_model, max_len=5000): """ Args