Machine-Learning
/
AIIA
1
0
Fork 0
Code Issues 1 Pull Requests Actions Packages Projects Releases Wiki Activity

Compare commits

base: Machine-Learning:ff6f279728c8eadbc77bda28969fa6e14949fd1b
Branches Tags
Machine-Learning:main
..
compare: Machine-Learning:7850cdba853bc01fd55a1e333a74144ed90a8fd1
Branches Tags
Machine-Learning:main

2 Commits
ff6f279728 ... 7850cdba85

Author SHA1 Message Date
Falko Victor Habel 7850cdba85 Merge pull request 'fixed saving' (#43) from feat/bugfix into main
Run VectorLoader Script / Explore-Gitea-Actions (push) Successful in 20s Details
Gitea Actions For AIIA / Explore-Gitea-Actions (push) Successful in 38s Details 
Reviewed-on: #43
 2025-04-24 07:37:45 +00:00
Falko Victor Habel 8c9853ade3 fixed saving
Gitea Actions For AIIA / Explore-Gitea-Actions (push) Successful in 45s Details
2025-04-24 09:37:19 +02:00
4 changed files with 4 additions and 4 deletions
Show Stats Expand all files Collapse all files

2
pyproject.toml
View File

@ -10,7 +10,7 @@ include = '\.pyi?$'
[project] [project]
name = "aiia" name = "aiia"
version = "0.3.2" version = "0.3.3"
description = "AIIA Deep Learning Model Implementation" description = "AIIA Deep Learning Model Implementation"
readme = "README.md" readme = "README.md"
authors = [ authors = [

2
setup.cfg
View File

@ -1,6 +1,6 @@
[metadata] [metadata]
name = aiia name = aiia
version = 0.3.2 version = 0.3.3
author = Falko Habel author = Falko Habel
author_email = falko.habel@gmx.de author_email = falko.habel@gmx.de
description = AIIA deep learning model implementation description = AIIA deep learning model implementation

2
src/aiia/__init__.py
View File

@ -4,4 +4,4 @@ from .data.DataLoader import DataLoader
from .pretrain.pretrainer import Pretrainer, ProjectionHead from .pretrain.pretrainer import Pretrainer, ProjectionHead
__version__ = "0.3.2" __version__ = "0.3.3"

2
src/aiia/pretrain/pretrainer.py
View File

@ -246,7 +246,7 @@ class Pretrainer:
if val_loss < best_val_loss: if val_loss < best_val_loss:
best_val_loss = val_loss best_val_loss = val_loss
self.model.save(output_path) self.model.save_pretrained(output_path)
print("Best model saved!") print("Best model saved!")
losses_path = os.path.join(os.path.dirname(output_path), 'losses.csv') losses_path = os.path.join(os.path.dirname(output_path), 'losses.csv')