Leacb4 commited on
Commit
bc323e6
·
verified ·
1 Parent(s): 01ca95a

Upload config.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.py +3 -4
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 = 32
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