| from pydantic import BaseModel | |
| class Config(BaseModel): | |
| clip_model_name: str = "sentence-transformers/clip-ViT-B-32" | |
| outfit_model_name: str = "elalber2000/clip-fashion-garments2look" | |
| minimum_similarity: float = 0.10 | |
| minimum_margin: float = 0.0 | |
| max_outfit_clothes: int = 4 | |
| local_emb_path: str = "/mnt/c/Users/ASUS/Downloads/clip-iqon-1hour-model/model" | |
| train_app_name: str = "garments-clip-contrastive" | |
| train_model_name: str = "sentence-transformers/clip-ViT-B-32" | |
| train_data_mnt: str = "/data" | |
| train_output_mnt: str = "/outputs" | |
| train_hf_cache_mnt: str = "/root/.cache/huggingface" | |
| train_gpu: str = "A10G" | |
| train_timeout: int = 70 * 60 | |
| train_memory: int = 32768 | |
| cnf = Config() |