Merge pull request 'develop' (#15 ) from develop into main

Reviewed-on: #15
Merge pull request 'bugfix' (#14 ) from feat/fix_saving into develop
2025-03-02 13:28:05 +00:00 · 2025-03-02 13:27:51 +00:00 · 2025-03-02 14:27:26 +01:00
3 changed files with 5 additions and 3 deletions
--- a/example.py
+++ b/example.py
@ -7,7 +7,7 @@ config = AIIAConfig(model_name="AIIA-Base-512x20k")
 model = AIIABase(config)

 # 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
 dataset_paths = [
--- a/requirements.txt
+++ b/requirements.txt
@ -3,3 +3,6 @@ numpy
 tqdm
 pytest
 pillow
+pandas
+torchvision
+pyarrow
--- a/src/aiia/model/Model.py
+++ b/src/aiia/model/Model.py
@ -130,7 +130,6 @@ class AIIABaseShared(AIIA):
        self.max_pool = nn.MaxPool2d(
            kernel_size=1,
            stride=1,
-            padding=1
        )

    def forward(self, x):
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