From 00424de105a7a10a9cbc35bcd37552f71e74b12d Mon Sep 17 00:00:00 2001 From: Jet Date: Wed, 21 Aug 2024 23:33:43 +0900 Subject: [PATCH] fix file extension --- wingman/wingman.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/wingman/wingman.py b/wingman/wingman.py index 214dd91..4d7e6e8 100644 --- 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"