| { | |
| "_name_or_path": "checkpoints/GR/Molecule3D_Gap_Small", | |
| "architectures": [ | |
| "GTMGCForGraphRegression" | |
| ], | |
| "atom_vocab_size": 513, | |
| "attn_drop": 0.0, | |
| "d_embed": 256, | |
| "d_ffn": 1024, | |
| "d_k": 256, | |
| "d_model": 256, | |
| "d_q": 256, | |
| "d_v": 256, | |
| "decoder_use_A_in_attn": true, | |
| "decoder_use_D_in_attn": true, | |
| "embed_style": "ogb", | |
| "encoder_use_A_in_attn": true, | |
| "encoder_use_D_in_attn": true, | |
| "ffn_drop": 0.0, | |
| "n_decode_layers": 6, | |
| "n_encode_layers": 6, | |
| "n_head": 8, | |
| "norm_drop": 0.0, | |
| "pre_ln": false, | |
| "qkv_bias": true, | |
| "torch_dtype": "float32", | |
| "transformers_version": "4.32.1" | |
| } | |