adelorenzo/stable
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add freqai.model_exists pytorch file type support

Browse Source
This commit is contained in:
Yinon Polak 2023-03-06 18:10:49 +02:00
parent 7eedcb9c14
commit 125085fbaf
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
freqtrade/freqai/freqai_interface.py
View File

@ -563,8 +563,11 @@ class IFreqaiModel(ABC):
file_type = ".joblib"
elif self.dd.model_type == 'keras':
file_type = ".h5"
elif 'stable_baselines' in self.dd.model_type or 'sb3_contrib' == self.dd.model_type:
elif ('stable_baselines' in self.dd.model_type or
'sb3_contrib' == self.dd.model_type or
'pytorch' == self.dd.model_type):
file_type = ".zip"
path_to_modelfile = Path(dk.data_path / f"{dk.model_filename}_model{file_type}")
file_exists = path_to_modelfile.is_file()
if file_exists: