Fix a typo

bioscan-ml · Dec 16, 2024 · 65520b0 · 65520b0
1 parent d244cec
commit 65520b0
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/...onfig/model_config/for_bioscan_5m/barcodeBERT_trained_with_5m/image_dna_text_seed_42.yaml b/...onfig/model_config/for_bioscan_5m/barcodeBERT_trained_with_5m/image_dna_text_seed_42.yaml
@@ -18,7 +18,7 @@ language:
 model_output_name: image_dna_text_4gpu
 evaluation_period: 1
 
-barcodeBERT_ckpt_path: ${project_root_path}/ckpt/barcodeBERT/trained_with_5m/(BIOSCAN-5M)-BEST_k4_6_6_w1_m0_r0.pt
+barcodeBERT_ckpt_path: ${project_root_path}/ckpt/barcodeBERT/trained_with_5m/BIOSCAN-5M-BEST_k4_6_6_w1_m0_r0.pt
 
 ckpt_path: ${project_root_path}/ckpt/bioscan_clip/new_5M_training/trained_with_5M_image_dna_text/best.pth
 output_dim: 768

diff --git a/bioscanclip/model/simple_clip.py b/bioscanclip/model/simple_clip.py
@@ -262,10 +262,12 @@ def load_clip_model(args, device=None):
         if args.model_config.dna.input_type == "sequence":
             if dna_model == "barcode_bert" or dna_model == "lora_barcode_bert":
                 barcode_BERT_ckpt = args.barcodeBERT_checkpoint
+                k = 5
                 if hasattr(args.model_config, "barcodeBERT_ckpt_path"):
                     barcode_BERT_ckpt = args.model_config.barcodeBERT_ckpt_path
+                    k=4
                 pre_trained_barcode_bert = load_pre_trained_bioscan_bert(
-                    bioscan_bert_checkpoint=barcode_BERT_ckpt)
+                    bioscan_bert_checkpoint=barcode_BERT_ckpt, k=k)
                 if disable_lora:
                     dna_encoder = LoRA_barcode_bert(model=pre_trained_barcode_bert, r=4,
                                                     num_classes=args.model_config.output_dim, lora_layer=[])