Debug.

bioscan-ml · Jul 24, 2024 · 0206a96 · 0206a96
1 parent e0ffe2b
commit 0206a96
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/bioscanclip/config/model_config/lora_vit_lora_barcode_bert_lora_bert_ssl_on_insect.yaml b/bioscanclip/config/model_config/lora_vit_lora_barcode_bert_lora_bert_ssl_on_insect.yaml
@@ -15,7 +15,7 @@ language:
   model: lora_bert
 
 model_output_name: trained_with_insect_image_dna_text
-evaluation_period: 40
+evaluation_period: 50
 ckpt_path: ${project_root_path}/ckpt/bioscan_clip/trained_with_bioscan_1m/image_dna_text.pth
 ckpt_trained_with_insect_image_dna_text_path: ${project_root_path}/ckpt/bioscan_clip/trained_with_insect_image_dna_text
 output_dim: 768

diff --git a/scripts/supervised_fine_tune_bioscan_clip_model_on_insect.py b/scripts/supervised_fine_tune_bioscan_clip_model_on_insect.py
@@ -21,7 +21,7 @@
 
 @hydra.main(config_path="../bioscanclip/config", config_name="global_config", version_base="1.1")
 def main(args: DictConfig) -> None:
-
+    os.environ["TOKENIZERS_PARALLELISM"] = "false"
     # world_size = torch.cuda.device_count()
     # print(f'world_size： {world_size}')
     # rank = 0