diff --git a/.gitmodules b/.gitmodules new file mode 100644 index 0000000..3174f6d --- /dev/null +++ b/.gitmodules @@ -0,0 +1,3 @@ +[submodule "Fine-Grained-ZSL-with-DNA"] + path = Fine-Grained-ZSL-with-DNA + url = git@github.com:zmgong/Fine-Grained-ZSL-with-DNA.git diff --git a/Fine-Grained-ZSL-with-DNA b/Fine-Grained-ZSL-with-DNA new file mode 160000 index 0000000..d966937 --- /dev/null +++ b/Fine-Grained-ZSL-with-DNA @@ -0,0 +1 @@ +Subproject commit d9669379e9fc3fb793fa1cfd0942da61d4fd7f18 diff --git a/scripts/supervised_fine_tune_bioscan_clip_model_on_insect.py b/scripts/supervised_fine_tune_bioscan_clip_model_on_insect.py index 1a6afa2..b229f2b 100644 --- a/scripts/supervised_fine_tune_bioscan_clip_model_on_insect.py +++ b/scripts/supervised_fine_tune_bioscan_clip_model_on_insect.py @@ -104,7 +104,7 @@ def main(args: DictConfig) -> None: args, world_size=None, rank=None, load_all_in_one=True) print("training...") - pbar = tqdm(range(args.model_config.epochs)) + pbar = tqdm(range(args.general_fine_tune_setting.epoch)) for epoch in pbar: pbar.set_description(f"Epoch: {epoch}") @@ -138,7 +138,7 @@ def main(args: DictConfig) -> None: print(f'Last image ckpt: {image_last_ckpt_path}') print(f'Last dna ckpt: {dna_last_ckpt_path}') # save_image_embedding to “image_embedding_from_bioscan_clip.csv” - folder_to_save_embed = os.path.join(args.project_root_path, "embedding_from_=bsc_fine_tuned_on_insect", + folder_to_save_embed = os.path.join(args.project_root_path, "embedding_from_bsc_fine_tuned_on_insect", formatted_datetime) os.makedirs(folder_to_save_embed, exist_ok=True) dna_embed_path = os.path.join(folder_to_save_embed, "dna_embedding_from_bioscan_clip.csv")