From ab7a474ab6e986062538bfbcfe44387c1177966b Mon Sep 17 00:00:00 2001 From: robcaulk Date: Sun, 14 May 2023 12:03:15 +0000 Subject: [PATCH] try limiting tb_logger to pytorch only (XGBoost still gets its callback) --- freqtrade/freqai/freqai_interface.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/freqtrade/freqai/freqai_interface.py b/freqtrade/freqai/freqai_interface.py index 8e34d1c28..ae16d3b99 100644 --- a/freqtrade/freqai/freqai_interface.py +++ b/freqtrade/freqai/freqai_interface.py @@ -635,9 +635,11 @@ class IFreqaiModel(ABC): dk.find_features(unfiltered_dataframe) dk.find_labels(unfiltered_dataframe) - self.tb_logger = TBLogger(dk.data_path) + if self.dd.model_type == "pytorch": + self.tb_logger = TBLogger(dk.data_path) model = self.train(unfiltered_dataframe, pair, dk) - self.tb_logger.close() + if self.dd.model_type == "pytorch": + self.tb_logger.close() self.dd.pair_dict[pair]["trained_timestamp"] = trained_timestamp dk.set_new_model_names(pair, trained_timestamp)