fix file extension

jjshoots · Aug 21, 2024 · 00424de · 00424de
1 parent 03aa00a
commit 00424de
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/wingman/wingman.py b/wingman/wingman.py
@@ -95,9 +95,9 @@ def __init__(
 
         # file paths
         self._model_file: Path = (
-            self._model_directory / f"weights{self._current_ckpt}.path"
+            self._model_directory / f"weights{self._current_ckpt}.pth"
         )
-        self._optim_file: Path = self._model_directory / "optimizer_path"
+        self._optim_file: Path = self._model_directory / "optimizer.pth"
         self._lowest_loss_file: Path = self._model_directory / "lowest_loss.npy"
         self._intermediary_file: Path = self._model_directory / "weights-1.npy"
         self._log_file: Path = self._model_directory / "log.txt"