try to avoid possible memory leaks

2022-12-07 20:08:31 +01:00
parent b438cd4b3f
commit 71c6ff18c4
2 changed files with 3 additions and 2 deletions
--- a/freqtrade/freqai/prediction_models/CNNPredictionModel.py
+++ b/freqtrade/freqai/prediction_models/CNNPredictionModel.py
@@ -133,6 +133,7 @@ class CNNPredictionModel(BaseTensorFlowModel):
        else:
            data = dk.data_dictionary["prediction_features"]
            data = tf.expand_dims(data, axis=0)
+            data = tf.convert_to_tensor(data)
            predictions = self.model(data, training=False)

        predictions = predictions[:, 0, 0]