We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 04a4a97 commit b8f5df7Copy full SHA for b8f5df7
1 file changed
freqtrade/freqai/torch/PyTorchModelTrainer.py
@@ -179,16 +179,12 @@ def save(self, path: Path):
179
path,
180
)
181
182
- def load(self, path: Path):
183
- checkpoint = torch.load(path)
184
- return self.load_from_checkpoint(checkpoint)
185
-
186
def load_from_checkpoint(self, checkpoint: dict):
187
"""
188
when using continual_learning, DataDrawer will load the dictionary
189
(containing state dicts and model_meta_data) by calling torch.load(path).
190
you can access this dict from any class that inherits IFreqaiModel by calling
191
- get_init_model method.
+ the get_init_model method.
192
193
self.model.load_state_dict(checkpoint["model_state_dict"])
194
self.optimizer.load_state_dict(checkpoint["optimizer_state_dict"])
0 commit comments