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

Compare commits

base: Machine-Learning:8d0548149840aef51dc625b8f5b3e8ef54bb21a8
Branches Tags
Machine-Learning:main
...
compare: Machine-Learning:40c2b6dd501a8f4dd3a7fca94118e968c900e9d9
Branches Tags
Machine-Learning:main

3 Commits
8d05481498 ... 40c2b6dd50

Author SHA1 Message Date
Falko Victor Habel 40c2b6dd50 Merge pull request 'develop' (#15) from develop into main 
Reviewed-on: #15
 2025-03-02 13:28:05 +00:00
Falko Victor Habel ee58b085f5 Merge pull request 'bugfix' (#14) from feat/fix_saving into develop 
Reviewed-on: #14
 2025-03-02 13:27:51 +00:00
Falko Victor Habel 66762775e9 bugfix 2025-03-02 14:27:26 +01:00
3 changed files with 5 additions and 3 deletions
Show Stats Expand all files Collapse all files

2
example.py
View File

@ -7,7 +7,7 @@ config = AIIAConfig(model_name="AIIA-Base-512x20k")
model = AIIABase(config) model = AIIABase(config)
# Initialize pretrainer with the model # Initialize pretrainer with the model
pretrainer = Pretrainer(model, learning_rate=1e-4) pretrainer = Pretrainer(model, learning_rate=1e-4, config=config)
# List of dataset paths # List of dataset paths
dataset_paths = [ dataset_paths = [

3
requirements.txt
View File

@ -3,3 +3,6 @@ numpy
tqdm tqdm
pytest pytest
pillow pillow
pandas
torchvision
pyarrow

1
src/aiia/model/Model.py
View File

@ -130,7 +130,6 @@ class AIIABaseShared(AIIA):
self.max_pool = nn.MaxPool2d( self.max_pool = nn.MaxPool2d(
kernel_size=1, kernel_size=1,
stride=1, stride=1,
padding=1
) )
def forward(self, x): def forward(self, x):