apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-05-15 11:02:36 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

add image datasets config

Browse source
This commit is contained in:
Yizhou Chi 2024-09-13 19:10:45 +08:00
parent cfa21ba27e
commit 7d8cb9afec
2 changed files with 27 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
expo/data/hf_data.py
View file

@ -115,8 +115,8 @@ class HFExpDataset(ExpDataset):
if __name__ == "__main__":
dataset_dir = "D:/work/automl/datasets"
save_analysis_pool = True
force_update = True
save_analysis_pool = False
force_update = False
datasets_dict = {"datasets": {}}
solution_designer = SolutionDesigner()
for dataset_meta in HFDATSETS:
@ -125,7 +125,7 @@ if __name__ == "__main__":
dataset_dir,
dataset_meta["dataset_name"],
target_col=dataset_meta["target_col"],
image_col=dataset_meta["image_col"],
image_col=dataset_meta.get("image_col", ""),
force_update=force_update,
modality=dataset_meta["modality"],
)