Set self.fitted = True instead of self._fitted.

microsoft · Jun 24, 2021 · 2ab4a9a · 2ab4a9a
1 parent 8ebdb1e
commit 2ab4a9a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/qlib/contrib/model/pytorch_nn.py b/qlib/contrib/model/pytorch_nn.py
@@ -297,7 +297,7 @@ def load(self, buffer, **kwargs):
             _model_path = os.path.join(model_dir, _model_name)
             # Load model
             self.dnn_model.load_state_dict(torch.load(_model_path))
-        self._fitted = True
+        self.fitted = True
 
 
 class AverageMeter: