Update model_checkpoint.py (#3801)

Lightning-AI · Oct 2, 2020 · 192fc01 · 192fc01
1 parent 7c6ed1f
commit 192fc01
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_lightning/callbacks/model_checkpoint.py b/pytorch_lightning/callbacks/model_checkpoint.py
@@ -549,5 +549,5 @@ def to_yaml(self, filepath: Optional[Union[str, Path]] = None):
         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)