diff --git a/example.py b/example.py
index 9e3b89f..8dbb67e 100644
--- 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 = [
diff --git a/requirements.txt b/requirements.txt
index 8db58b1..8e2d666 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,4 +2,7 @@ torch>=2.5.0
 numpy
 tqdm
 pytest
-pillow
\ No newline at end of file
+pillow
+pandas
+torchvision
+pyarrow
\ No newline at end of file
diff --git a/src/aiia/model/Model.py b/src/aiia/model/Model.py
index c067a67..39d5684 100644
--- 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):