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

Merge pull request 'bugfix' (#14) from feat/fix_saving into develop 

Reviewed-on: #14
This commit is contained in:
Falko Victor Habel 2025-03-02 13:27:51 +00:00
parent 8d05481498 66762775e9
commit ee58b085f5
3 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File 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 = [

5
requirements.txt
View File

@ -2,4 +2,7 @@ torch>=2.5.0
numpy 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):