|
{ |
|
"architectures": [ |
|
"StructurePrediction" |
|
], |
|
"depth_resnet": 5, |
|
"enable_cross": true, |
|
"hidden_size": 1024, |
|
"key_chunk_size_seq": 1024, |
|
"key_chunk_size_smiles": 1024, |
|
"linear_mem_attn": true, |
|
"model_type": "bert", |
|
"n_cross_attention": 3, |
|
"num_atoms": 14, |
|
"num_embeddings": 30, |
|
"num_ipa_layers": 8, |
|
"num_rigid_groups": 8, |
|
"query_chunk_size_seq": 512, |
|
"query_chunk_size_smiles": 512, |
|
"seq_config": { |
|
"_name_or_path": "", |
|
"add_cross_attention": false, |
|
"architectures": [ |
|
"BertForMaskedLM" |
|
], |
|
"attention_probs_dropout_prob": 0.0, |
|
"bad_words_ids": null, |
|
"bos_token_id": null, |
|
"chunk_size_feed_forward": 0, |
|
"classifier_dropout": null, |
|
"cross_attention_hidden_size": null, |
|
"decoder_start_token_id": null, |
|
"diversity_penalty": 0.0, |
|
"do_sample": false, |
|
"early_stopping": false, |
|
"encoder_no_repeat_ngram_size": 0, |
|
"eos_token_id": null, |
|
"exponential_decay_length_penalty": null, |
|
"finetuning_task": null, |
|
"forced_bos_token_id": null, |
|
"forced_eos_token_id": null, |
|
"hidden_act": "gelu", |
|
"hidden_dropout_prob": 0.0, |
|
"hidden_size": 1024, |
|
"id2label": { |
|
"0": "LABEL_0", |
|
"1": "LABEL_1" |
|
}, |
|
"initializer_range": 0.02, |
|
"intermediate_size": 4096, |
|
"is_decoder": false, |
|
"is_encoder_decoder": false, |
|
"label2id": { |
|
"LABEL_0": 0, |
|
"LABEL_1": 1 |
|
}, |
|
"layer_norm_eps": 1e-12, |
|
"length_penalty": 1.0, |
|
"max_length": 20, |
|
"max_position_embeddings": 40000, |
|
"min_length": 0, |
|
"model_type": "bert", |
|
"no_repeat_ngram_size": 0, |
|
"num_attention_heads": 16, |
|
"num_beam_groups": 1, |
|
"num_beams": 1, |
|
"num_hidden_layers": 30, |
|
"num_return_sequences": 1, |
|
"output_attentions": false, |
|
"output_hidden_states": false, |
|
"output_scores": false, |
|
"pad_token_id": 0, |
|
"position_embedding_type": "absolute", |
|
"prefix": null, |
|
"problem_type": null, |
|
"pruned_heads": {}, |
|
"remove_invalid_values": false, |
|
"repetition_penalty": 1.0, |
|
"return_dict": true, |
|
"return_dict_in_generate": false, |
|
"sep_token_id": null, |
|
"task_specific_params": null, |
|
"temperature": 1.0, |
|
"tie_encoder_decoder": false, |
|
"tie_word_embeddings": true, |
|
"tokenizer_class": null, |
|
"top_k": 50, |
|
"top_p": 1.0, |
|
"torch_dtype": null, |
|
"torchscript": false, |
|
"transformers_version": "4.21.0.dev0", |
|
"type_vocab_size": 2, |
|
"typical_p": 1.0, |
|
"use_bfloat16": false, |
|
"use_cache": true, |
|
"vocab_size": 30 |
|
}, |
|
"seq_ipa_config": { |
|
"_name_or_path": "", |
|
"add_cross_attention": false, |
|
"architectures": null, |
|
"bad_words_ids": null, |
|
"bert_config": { |
|
"_name_or_path": "", |
|
"add_cross_attention": false, |
|
"architectures": [ |
|
"BertForMaskedLM" |
|
], |
|
"attention_probs_dropout_prob": 0.0, |
|
"bad_words_ids": null, |
|
"bos_token_id": null, |
|
"chunk_size_feed_forward": 0, |
|
"classifier_dropout": null, |
|
"cross_attention_hidden_size": null, |
|
"decoder_start_token_id": null, |
|
"diversity_penalty": 0.0, |
|
"do_sample": false, |
|
"early_stopping": false, |
|
"encoder_no_repeat_ngram_size": 0, |
|
"eos_token_id": null, |
|
"exponential_decay_length_penalty": null, |
|
"finetuning_task": null, |
|
"forced_bos_token_id": null, |
|
"forced_eos_token_id": null, |
|
"hidden_act": "gelu", |
|
"hidden_dropout_prob": 0.0, |
|
"hidden_size": 1024, |
|
"id2label": { |
|
"0": "LABEL_0", |
|
"1": "LABEL_1" |
|
}, |
|
"initializer_range": 0.02, |
|
"intermediate_size": 4096, |
|
"is_decoder": false, |
|
"is_encoder_decoder": false, |
|
"label2id": { |
|
"LABEL_0": 0, |
|
"LABEL_1": 1 |
|
}, |
|
"layer_norm_eps": 1e-12, |
|
"length_penalty": 1.0, |
|
"max_length": 20, |
|
"max_position_embeddings": 40000, |
|
"min_length": 0, |
|
"model_type": "bert", |
|
"no_repeat_ngram_size": 0, |
|
"num_attention_heads": 16, |
|
"num_beam_groups": 1, |
|
"num_beams": 1, |
|
"num_hidden_layers": 30, |
|
"num_return_sequences": 1, |
|
"output_attentions": false, |
|
"output_hidden_states": false, |
|
"output_scores": false, |
|
"pad_token_id": 0, |
|
"position_embedding_type": "absolute", |
|
"prefix": null, |
|
"problem_type": null, |
|
"pruned_heads": {}, |
|
"remove_invalid_values": false, |
|
"repetition_penalty": 1.0, |
|
"return_dict": true, |
|
"return_dict_in_generate": false, |
|
"sep_token_id": null, |
|
"task_specific_params": null, |
|
"temperature": 1.0, |
|
"tie_encoder_decoder": false, |
|
"tie_word_embeddings": true, |
|
"tokenizer_class": null, |
|
"top_k": 50, |
|
"top_p": 1.0, |
|
"torch_dtype": null, |
|
"torchscript": false, |
|
"transformers_version": "4.21.0.dev0", |
|
"type_vocab_size": 2, |
|
"typical_p": 1.0, |
|
"use_bfloat16": false, |
|
"use_cache": true, |
|
"vocab_size": 30 |
|
}, |
|
"bos_token_id": null, |
|
"chunk_size_feed_forward": 0, |
|
"cross_attention_hidden_size": null, |
|
"decoder_start_token_id": null, |
|
"diversity_penalty": 0.0, |
|
"do_sample": false, |
|
"early_stopping": false, |
|
"encoder_no_repeat_ngram_size": 0, |
|
"eos_token_id": null, |
|
"exponential_decay_length_penalty": null, |
|
"finetuning_task": null, |
|
"forced_bos_token_id": null, |
|
"forced_eos_token_id": null, |
|
"id2label": { |
|
"0": "LABEL_0", |
|
"1": "LABEL_1" |
|
}, |
|
"is_decoder": false, |
|
"is_encoder_decoder": false, |
|
"key_chunk_size": 1024, |
|
"label2id": { |
|
"LABEL_0": 0, |
|
"LABEL_1": 1 |
|
}, |
|
"length_penalty": 1.0, |
|
"max_length": 20, |
|
"min_length": 0, |
|
"model_type": "", |
|
"no_repeat_ngram_size": 0, |
|
"num_beam_groups": 1, |
|
"num_beams": 1, |
|
"num_intermediate_layers": 3, |
|
"num_ipa_heads": 16, |
|
"num_points": 8, |
|
"num_return_sequences": 1, |
|
"output_attentions": false, |
|
"output_hidden_states": false, |
|
"output_scores": false, |
|
"pad_token_id": null, |
|
"prefix": null, |
|
"problem_type": null, |
|
"pruned_heads": {}, |
|
"query_chunk_size": 512, |
|
"remove_invalid_values": false, |
|
"repetition_penalty": 1.0, |
|
"return_dict": true, |
|
"return_dict_in_generate": false, |
|
"sep_token_id": null, |
|
"task_specific_params": null, |
|
"temperature": 1.0, |
|
"tie_encoder_decoder": false, |
|
"tie_word_embeddings": true, |
|
"tokenizer_class": null, |
|
"top_k": 50, |
|
"top_p": 1.0, |
|
"torch_dtype": null, |
|
"torchscript": false, |
|
"transformers_version": "4.21.0.dev0", |
|
"typical_p": 1.0, |
|
"use_bfloat16": false |
|
}, |
|
"seq_vocab": { |
|
"A": 6, |
|
"B": 27, |
|
"C": 23, |
|
"D": 14, |
|
"E": 9, |
|
"F": 19, |
|
"G": 7, |
|
"H": 22, |
|
"I": 11, |
|
"K": 12, |
|
"L": 5, |
|
"M": 21, |
|
"N": 17, |
|
"O": 29, |
|
"P": 16, |
|
"Q": 18, |
|
"R": 13, |
|
"S": 10, |
|
"T": 15, |
|
"U": 26, |
|
"V": 8, |
|
"W": 24, |
|
"X": 25, |
|
"Y": 20, |
|
"Z": 28, |
|
"[CLS]": 2, |
|
"[MASK]": 4, |
|
"[PAD]": 0, |
|
"[SEP]": 3, |
|
"[UNK]": 1 |
|
}, |
|
"smiles_config": { |
|
"_name_or_path": "", |
|
"add_cross_attention": false, |
|
"architectures": [ |
|
"BertForMaskedLM" |
|
], |
|
"attention_probs_dropout_prob": 0.1, |
|
"bad_words_ids": null, |
|
"bos_token_id": null, |
|
"chunk_size_feed_forward": 0, |
|
"classifier_dropout": null, |
|
"cross_attention_hidden_size": null, |
|
"decoder_start_token_id": null, |
|
"diversity_penalty": 0.0, |
|
"do_sample": false, |
|
"early_stopping": false, |
|
"encoder_no_repeat_ngram_size": 0, |
|
"eos_token_id": null, |
|
"exponential_decay_length_penalty": null, |
|
"finetuning_task": null, |
|
"forced_bos_token_id": null, |
|
"forced_eos_token_id": null, |
|
"gradient_checkpointing": false, |
|
"hidden_act": "gelu", |
|
"hidden_dropout_prob": 0.1, |
|
"hidden_size": 768, |
|
"id2label": { |
|
"0": "LABEL_0", |
|
"1": "LABEL_1" |
|
}, |
|
"initializer_range": 0.02, |
|
"intermediate_size": 3072, |
|
"is_decoder": false, |
|
"is_encoder_decoder": false, |
|
"label2id": { |
|
"LABEL_0": 0, |
|
"LABEL_1": 1 |
|
}, |
|
"layer_norm_eps": 1e-12, |
|
"length_penalty": 1.0, |
|
"max_length": 20, |
|
"max_position_embeddings": 512, |
|
"min_length": 0, |
|
"model_type": "bert", |
|
"no_repeat_ngram_size": 0, |
|
"num_attention_heads": 12, |
|
"num_beam_groups": 1, |
|
"num_beams": 1, |
|
"num_hidden_layers": 12, |
|
"num_return_sequences": 1, |
|
"output_attentions": false, |
|
"output_hidden_states": false, |
|
"output_scores": false, |
|
"pad_token_id": 0, |
|
"position_embedding_type": "absolute", |
|
"prefix": null, |
|
"problem_type": null, |
|
"pruned_heads": {}, |
|
"remove_invalid_values": false, |
|
"repetition_penalty": 1.0, |
|
"return_dict": true, |
|
"return_dict_in_generate": false, |
|
"sep_token_id": null, |
|
"task_specific_params": null, |
|
"temperature": 1.0, |
|
"tie_encoder_decoder": false, |
|
"tie_word_embeddings": true, |
|
"tokenizer_class": null, |
|
"top_k": 50, |
|
"top_p": 1.0, |
|
"torch_dtype": null, |
|
"torchscript": false, |
|
"transformers_version": "4.21.0.dev0", |
|
"type_vocab_size": 2, |
|
"typical_p": 1.0, |
|
"use_bfloat16": false, |
|
"use_cache": true, |
|
"vocab_size": 30000 |
|
}, |
|
"smiles_ipa_config": { |
|
"_name_or_path": "", |
|
"add_cross_attention": false, |
|
"architectures": null, |
|
"bad_words_ids": null, |
|
"bert_config": { |
|
"_name_or_path": "", |
|
"add_cross_attention": false, |
|
"architectures": [ |
|
"BertForMaskedLM" |
|
], |
|
"attention_probs_dropout_prob": 0.1, |
|
"bad_words_ids": null, |
|
"bos_token_id": null, |
|
"chunk_size_feed_forward": 0, |
|
"classifier_dropout": null, |
|
"cross_attention_hidden_size": null, |
|
"decoder_start_token_id": null, |
|
"diversity_penalty": 0.0, |
|
"do_sample": false, |
|
"early_stopping": false, |
|
"encoder_no_repeat_ngram_size": 0, |
|
"eos_token_id": null, |
|
"exponential_decay_length_penalty": null, |
|
"finetuning_task": null, |
|
"forced_bos_token_id": null, |
|
"forced_eos_token_id": null, |
|
"gradient_checkpointing": false, |
|
"hidden_act": "gelu", |
|
"hidden_dropout_prob": 0.1, |
|
"hidden_size": 768, |
|
"id2label": { |
|
"0": "LABEL_0", |
|
"1": "LABEL_1" |
|
}, |
|
"initializer_range": 0.02, |
|
"intermediate_size": 3072, |
|
"is_decoder": false, |
|
"is_encoder_decoder": false, |
|
"label2id": { |
|
"LABEL_0": 0, |
|
"LABEL_1": 1 |
|
}, |
|
"layer_norm_eps": 1e-12, |
|
"length_penalty": 1.0, |
|
"max_length": 20, |
|
"max_position_embeddings": 512, |
|
"min_length": 0, |
|
"model_type": "bert", |
|
"no_repeat_ngram_size": 0, |
|
"num_attention_heads": 12, |
|
"num_beam_groups": 1, |
|
"num_beams": 1, |
|
"num_hidden_layers": 12, |
|
"num_return_sequences": 1, |
|
"output_attentions": false, |
|
"output_hidden_states": false, |
|
"output_scores": false, |
|
"pad_token_id": 0, |
|
"position_embedding_type": "absolute", |
|
"prefix": null, |
|
"problem_type": null, |
|
"pruned_heads": {}, |
|
"remove_invalid_values": false, |
|
"repetition_penalty": 1.0, |
|
"return_dict": true, |
|
"return_dict_in_generate": false, |
|
"sep_token_id": null, |
|
"task_specific_params": null, |
|
"temperature": 1.0, |
|
"tie_encoder_decoder": false, |
|
"tie_word_embeddings": true, |
|
"tokenizer_class": null, |
|
"top_k": 50, |
|
"top_p": 1.0, |
|
"torch_dtype": null, |
|
"torchscript": false, |
|
"transformers_version": "4.21.0.dev0", |
|
"type_vocab_size": 2, |
|
"typical_p": 1.0, |
|
"use_bfloat16": false, |
|
"use_cache": true, |
|
"vocab_size": 30000 |
|
}, |
|
"bos_token_id": null, |
|
"chunk_size_feed_forward": 0, |
|
"cross_attention_hidden_size": null, |
|
"decoder_start_token_id": null, |
|
"diversity_penalty": 0.0, |
|
"do_sample": false, |
|
"early_stopping": false, |
|
"encoder_no_repeat_ngram_size": 0, |
|
"eos_token_id": null, |
|
"exponential_decay_length_penalty": null, |
|
"finetuning_task": null, |
|
"forced_bos_token_id": null, |
|
"forced_eos_token_id": null, |
|
"id2label": { |
|
"0": "LABEL_0", |
|
"1": "LABEL_1" |
|
}, |
|
"is_decoder": false, |
|
"is_encoder_decoder": false, |
|
"key_chunk_size": 1024, |
|
"label2id": { |
|
"LABEL_0": 0, |
|
"LABEL_1": 1 |
|
}, |
|
"length_penalty": 1.0, |
|
"max_length": 20, |
|
"min_length": 0, |
|
"model_type": "", |
|
"no_repeat_ngram_size": 0, |
|
"num_beam_groups": 1, |
|
"num_beams": 1, |
|
"num_intermediate_layers": 3, |
|
"num_ipa_heads": 12, |
|
"num_points": 8, |
|
"num_return_sequences": 1, |
|
"output_attentions": false, |
|
"output_hidden_states": false, |
|
"output_scores": false, |
|
"pad_token_id": null, |
|
"prefix": null, |
|
"problem_type": null, |
|
"pruned_heads": {}, |
|
"query_chunk_size": 512, |
|
"remove_invalid_values": false, |
|
"repetition_penalty": 1.0, |
|
"return_dict": true, |
|
"return_dict_in_generate": false, |
|
"sep_token_id": null, |
|
"task_specific_params": null, |
|
"temperature": 1.0, |
|
"tie_encoder_decoder": false, |
|
"tie_word_embeddings": true, |
|
"tokenizer_class": null, |
|
"top_k": 50, |
|
"top_p": 1.0, |
|
"torch_dtype": null, |
|
"torchscript": false, |
|
"transformers_version": "4.21.0.dev0", |
|
"typical_p": 1.0, |
|
"use_bfloat16": false |
|
}, |
|
"torch_dtype": "float32", |
|
"transformers_version": "4.21.0.dev0", |
|
"width_resnet": 128 |
|
} |
|
|