add get_last_checkpoint() (#883)

Summary:
Pull Request resolved: #883

function to fetch the last checkpoint

Reviewed By: diego-urgell

Differential Revision: D61409516

fbshipit-source-id: 72f991cc9a5a3d530bad955d125df8cb039323e7

Author: Richard Wang

torchtnt/utils/checkpoint.py

@@ -477,6 +477,14 @@ def should_save_checkpoint(self, checkpoint: CheckpointPath) -> bool:
             self._ckpt_paths[0], mode=best_checkpoint_config.mode
         )
 
+    def get_last_checkpoint(self) -> Optional[CheckpointPath]:
+        """
+        Returns last checkpoint to be added
+        """
+        if len(self._ckpt_paths) == 0:
+            return None
+        return self._ckpt_paths[-1]
+
     def append_checkpoint(self, ckpt: CheckpointPath) -> None:
         """
         This will update the internal state to keep track of the checkpoint. This function should only be called