adding checkpoints

method/custom_clip.py

@@ -682,7 +682,7 @@ def create_model_and_transforms(
         image_mean: Optional[Tuple[float, ...]] = None,
         image_std: Optional[Tuple[float, ...]] = None,
         aug_cfg: Optional[Union[Dict[str, Any], AugmentationCfg]] = None,
-        cache_dir: Optional[str] = None,
+        cache_dir: Optional[str] = "weights/ViT-L-14-336px.pt",
         output_dict: Optional[bool] = None,
 ):
     ######### create the clip model

run.sh

@@ -1,6 +1,6 @@
 #!/bin/bash
 
-ckt_path="weights/pretrained_mvtec_colondb.pth"
+ckt_path="weights/baseline.pth"
 gpu_id=0
 
 CUDA_VISIBLE_DEVICES=$gpu_id python test.py --ckt_path $ckt_path --testing_data mvtec

weights/ViT-L-14-336px.pt/ViT-L-14-336px.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3035c92b350959924f9f00213499208652fc7ea050643e8b385c2dac08641f02
+size 934088680

weights/baseline.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:be51a42c052bd4cf060e54f503a1f5d0b2a3b899bc8dc2e243042f18b215427e
+size 42673907