{
  "_name_or_path": "checkpoints/GR/Molecule3D_Gap_Large",
  "architectures": [
    "GTMGCForGraphRegression"
  ],
  "atom_vocab_size": 513,
  "attn_drop": 0.0,
  "d_embed": 512,
  "d_ffn": 2048,
  "d_k": 512,
  "d_model": 512,
  "d_q": 512,
  "d_v": 512,
  "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": 8,
  "n_head": 32,
  "norm_drop": 0.0,
  "pre_ln": false,
  "qkv_bias": true,
  "torch_dtype": "float32",
  "transformers_version": "4.32.1"
}