Spaces:
Runtime error
Runtime error
| artifacts_root: artifacts | |
| data_ingestion: | |
| root_dir: artifacts/data_ingestion | |
| dataset_name: "HuggingFaceM4/FairFace" | |
| dataset_config: "0.25" | |
| local_data_dir: artifacts/data_ingestion/dataset | |
| data_preparation: | |
| root_dir: artifacts/data_preparation | |
| # Path to the raw dataset downloaded by the previous stage | |
| raw_data_path: artifacts/data_ingestion/dataset | |
| # Path where we will save the cleaned dataframe | |
| cleaned_data_path: artifacts/data_preparation/fairface_cleaned.csv | |
| multi_task_model_trainer: | |
| root_dir: artifacts/multi_task_model_trainer | |
| # The path to our cleaned CSV file from the previous stage | |
| data_path: artifacts/data_preparation/fairface_cleaned.csv | |
| # Where to save the final multi-task model | |
| trained_model_path: artifacts/multi_task_model_trainer/facial_demographics_model | |
| # The base model from Hugging Face | |
| model_name: "google/efficientnet-b2" | |