Machine-Learning
/
AIIA
1
0
Fork 0
Code Issues 1 Pull Requests Actions Packages Projects Releases Wiki Activity

increased epochs

This commit is contained in:
Falko Victor Habel 2025-01-27 11:06:42 +01:00
parent 8d08bfc14c
commit 102d952043
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/pretrain.py
View File

@ -223,4 +223,4 @@ def pretrain_model(data_path1, data_path2, num_epochs=3):
if __name__ == "__main__": if __name__ == "__main__":
data_path1 = "/root/training_data/vision-dataset/images_checkpoint.parquet" data_path1 = "/root/training_data/vision-dataset/images_checkpoint.parquet"
data_path2 = "/root/training_data/vision-dataset/vec_images_dataset.parquet" data_path2 = "/root/training_data/vision-dataset/vec_images_dataset.parquet"
pretrain_model(data_path1, data_path2, num_epochs=3) pretrain_model(data_path1, data_path2, num_epochs=10)