From 102d9520437ff031bc994548ac9eefbe4c79b71e Mon Sep 17 00:00:00 2001 From: Falko Habel Date: Mon, 27 Jan 2025 11:06:42 +0100 Subject: [PATCH] increased epochs --- src/pretrain.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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