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

Compare commits

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

No commits in common. "7850cdba853bc01fd55a1e333a74144ed90a8fd1" and "ff6f279728c8eadbc77bda28969fa6e14949fd1b" have entirely different histories.
7850cdba85 ... ff6f279728

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.3" version = "0.3.2"
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.3 version = 0.3.2
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.3" __version__ = "0.3.2"

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_pretrained(output_path) self.model.save(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')