|
|
name: bird_classifier |
|
|
|
|
|
train_data: "/home/jun/work/species_genAI/finetune/pretrain_species_encoder_susbsystems/57birds/large_species_img_w_prot_list_50birds_subsystems_resize_512.train" |
|
|
valid_data: "/home/jun/work/species_genAI/finetune/pretrain_species_encoder_susbsystems/57birds/large_species_img_w_prot_list_50birds_subsystems_resize_512.validate" |
|
|
|
|
|
|
|
|
embeddings: "/home/jun/work/species_genAI/embeddings_prot_bert_bfd_latin_name" |
|
|
species_classes: "/home/jun/work/species_genAI/finetune/pretrain_species_encoder_susbsystems/57birds/57_species_id2.json" |
|
|
vocabulary: "/home/jun/work/species_genAI/finetune/pretrain_species_encoder_susbsystems/57birds/57_bird_species_protein_vocabulary.txt" |
|
|
|
|
|
max_tokens: 6000 |
|
|
|
|
|
|
|
|
steps: 500000 |
|
|
batch_size: 16 |
|
|
num_workers: 8 |
|
|
learning_rate: 0.0003 |
|
|
accumulate_grad_batches: 1 |
|
|
early_stop_callback: 1000 |
|
|
|
|
|
classify: True |
|
|
|
|
|
gpus: 1 |
|
|
precision: 16-mixed |
|
|
|
