diff --git a/image-inpainting/results/testset/tikaiz-16.9248.npz b/image-inpainting/results/testset/tikaiz-16.9248.npz
new file mode 100644
index 0000000..5db35ba
Binary files /dev/null and b/image-inpainting/results/testset/tikaiz-16.9248.npz differ
diff --git a/image-inpainting/results/testset/tikaiz-17.3305.npz b/image-inpainting/results/testset/tikaiz-17.3305.npz
new file mode 100644
index 0000000..67b0195
Binary files /dev/null and b/image-inpainting/results/testset/tikaiz-17.3305.npz differ
diff --git a/image-inpainting/src/__pycache__/architecture.cpython-313.pyc b/image-inpainting/src/__pycache__/architecture.cpython-313.pyc
index cac7a2a..1a31522 100644
Binary files a/image-inpainting/src/__pycache__/architecture.cpython-313.pyc and b/image-inpainting/src/__pycache__/architecture.cpython-313.pyc differ
diff --git a/image-inpainting/src/__pycache__/datasets.cpython-313.pyc b/image-inpainting/src/__pycache__/datasets.cpython-313.pyc
index ded87ad..cfb947f 100644
Binary files a/image-inpainting/src/__pycache__/datasets.cpython-313.pyc and b/image-inpainting/src/__pycache__/datasets.cpython-313.pyc differ
diff --git a/image-inpainting/src/__pycache__/train.cpython-313.pyc b/image-inpainting/src/__pycache__/train.cpython-313.pyc
index baa1f65..e8ef8b2 100644
Binary files a/image-inpainting/src/__pycache__/train.cpython-313.pyc and b/image-inpainting/src/__pycache__/train.cpython-313.pyc differ
diff --git a/image-inpainting/src/main.py b/image-inpainting/src/main.py
index da50793..761166e 100644
--- a/image-inpainting/src/main.py
+++ b/image-inpainting/src/main.py
@@ -24,17 +24,17 @@ if __name__ == '__main__':
     config_dict['results_path'] = os.path.join(project_root, "results")
     config_dict['data_path'] = os.path.join(project_root, "data", "dataset")
     config_dict['device'] = None
-    config_dict['learningrate'] = 8e-4  # Higher max LR for OneCycleLR
+    config_dict['learningrate'] = 1e-3  # Higher max LR for OneCycleLR
     config_dict['weight_decay'] = 1e-5  # Lower for faster learning
     config_dict['n_updates'] = 3500  # Reduced for fast training
-    config_dict['batchsize'] = 16  # Larger batch for speed
-    config_dict['early_stopping_patience'] = 12  # Adjusted patience
+    config_dict['batchsize'] = 96  # Larger batch for speed
+    config_dict['early_stopping_patience'] = 3  # Adjusted patience
     config_dict['use_wandb'] = False
 
     config_dict['print_train_stats_at'] = 10
     config_dict['print_stats_at'] = 100
     config_dict['plot_at'] = 500
-    config_dict['validate_at'] = 150  # More frequent validation
+    config_dict['validate_at'] = 500  # More frequent validation
 
     network_config = {
         'n_in_channels': 4,