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

furhter batchsizing downsizing

This commit is contained in:
Falko Victor Habel 2025-01-27 09:08:39 +01:00
parent 8dad1d7150
commit b546f4ee27
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

@ -49,7 +49,7 @@ def pretrain_model(data_path1, data_path2, num_epochs=3):
aiia_loader = AIIADataLoader( aiia_loader = AIIADataLoader(
merged_df, merged_df,
column="image_bytes", column="image_bytes",
batch_size=8, batch_size=4,
pretraining=True, pretraining=True,
collate_fn=safe_collate collate_fn=safe_collate
) )