Upload config.py with huggingface_hub
Browse files
config.py
CHANGED
|
@@ -30,7 +30,7 @@ hierarchy_emb_dim = 64
|
|
| 30 |
main_emb_dim = 512
|
| 31 |
|
| 32 |
# Default training hyperparameters
|
| 33 |
-
DEFAULT_BATCH_SIZE =
|
| 34 |
DEFAULT_LEARNING_RATE = 1.5e-5
|
| 35 |
DEFAULT_TEMPERATURE = 0.09
|
| 36 |
|
|
@@ -46,11 +46,11 @@ local_dataset_path = str(ROOT_DIR / "data" / "data.csv")
|
|
| 46 |
color_model_path = str(ROOT_DIR / "models" / "color_model.pt")
|
| 47 |
hierarchy_model_path = str(ROOT_DIR / "models" / "hierarchy_model.pth")
|
| 48 |
main_model_path = str(ROOT_DIR / "models" / "gap_clip.pth")
|
| 49 |
-
tokeniser_path = str(ROOT_DIR / "tokenizer_vocab.json")
|
| 50 |
images_dir = str(ROOT_DIR / "data" / "images")
|
| 51 |
fashion_mnist_csv = str(ROOT_DIR / "data" / "fashion-mnist_test.csv")
|
| 52 |
|
| 53 |
|
|
|
|
| 54 |
def print_config() -> None:
|
| 55 |
"""Pretty-print core configuration."""
|
| 56 |
print("GAP-CLIP Configuration")
|
|
@@ -68,8 +68,7 @@ def validate_paths() -> dict[str, bool]:
|
|
| 68 |
"local_dataset_path": Path(local_dataset_path).exists(),
|
| 69 |
"color_model_path": Path(color_model_path).exists(),
|
| 70 |
"hierarchy_model_path": Path(hierarchy_model_path).exists(),
|
| 71 |
-
"main_model_path": Path(main_model_path).exists()
|
| 72 |
-
"tokeniser_path": Path(tokeniser_path).exists(),
|
| 73 |
}
|
| 74 |
return checks
|
| 75 |
|
|
|
|
| 30 |
main_emb_dim = 512
|
| 31 |
|
| 32 |
# Default training hyperparameters
|
| 33 |
+
DEFAULT_BATCH_SIZE = 64
|
| 34 |
DEFAULT_LEARNING_RATE = 1.5e-5
|
| 35 |
DEFAULT_TEMPERATURE = 0.09
|
| 36 |
|
|
|
|
| 46 |
color_model_path = str(ROOT_DIR / "models" / "color_model.pt")
|
| 47 |
hierarchy_model_path = str(ROOT_DIR / "models" / "hierarchy_model.pth")
|
| 48 |
main_model_path = str(ROOT_DIR / "models" / "gap_clip.pth")
|
|
|
|
| 49 |
images_dir = str(ROOT_DIR / "data" / "images")
|
| 50 |
fashion_mnist_csv = str(ROOT_DIR / "data" / "fashion-mnist_test.csv")
|
| 51 |
|
| 52 |
|
| 53 |
+
|
| 54 |
def print_config() -> None:
|
| 55 |
"""Pretty-print core configuration."""
|
| 56 |
print("GAP-CLIP Configuration")
|
|
|
|
| 68 |
"local_dataset_path": Path(local_dataset_path).exists(),
|
| 69 |
"color_model_path": Path(color_model_path).exists(),
|
| 70 |
"hierarchy_model_path": Path(hierarchy_model_path).exists(),
|
| 71 |
+
"main_model_path": Path(main_model_path).exists()
|
|
|
|
| 72 |
}
|
| 73 |
return checks
|
| 74 |
|