diff --git a/src/pretrain.py b/src/pretrain.py
index 6e4c05c..02e4e7f 100644
--- a/src/pretrain.py
+++ b/src/pretrain.py
@@ -223,4 +223,4 @@ def pretrain_model(data_path1, data_path2, num_epochs=3):
 if __name__ == "__main__":
     data_path1 = "/root/training_data/vision-dataset/images_checkpoint.parquet"
     data_path2 = "/root/training_data/vision-dataset/vec_images_dataset.parquet"
-    pretrain_model(data_path1, data_path2, num_epochs=3)
\ No newline at end of file
+    pretrain_model(data_path1, data_path2, num_epochs=10)
\ No newline at end of file