Update model_checkpoint.py (#3801)

2020-10-02 11:49:46 -07:00 · 2020-10-02 11:49:46 -07:00 · 192fc018f3
parent 7c6ed1fa28
commit 192fc018f3
1 changed files with 1 additions and 1 deletions
--- a/pytorch_lightning/callbacks/model_checkpoint.py
+++ b/pytorch_lightning/callbacks/model_checkpoint.py
@ -549,5 +549,5 @@ class ModelCheckpoint(Callback):
        best_k = {k: v.item() for k, v in self.best_k_models.items()}
        if filepath is None:
            filepath = os.path.join(self.dirpath, "best_k_models.yaml")
-        with open(filepath, "w") as fp:
+        with self._fs.open(filepath, "w") as fp:
            yaml.dump(best_k, fp)