Tinsae/Finetuning
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

added loading and testing the model again

Browse Source
This commit is contained in:
Tinsae
2026-05-05 22:43:57 +02:00
parent cfa4da697e
commit a4191fa00c
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
architectures/Model.py
View File

@@ -16,7 +16,6 @@ class Model(ABC):
@abstractmethod @abstractmethod
def get(self): def get(self):
# return the model
pass pass
def train(self, epochs, loader, rate): def train(self, epochs, loader, rate):
@@ -66,7 +65,7 @@ class Model(ABC):
save_dir = Path("trained_models") save_dir = Path("trained_models")
save_dir.mkdir(parents=True, exist_ok=True) save_dir.mkdir(parents=True, exist_ok=True)
# 2. Determine filename (Default to class name if not provided) # Determine filename (Default to class name if not provided)
if filename is None: if filename is None:
filename = f"{self.__class__.__name__.lower()}.pth" filename = f"{self.__class__.__name__.lower()}.pth"