jakesmith-microsoft's picture
Replace with v2 weights (#7)
8394616
Raw History Blame Contribute Delete
1.22 kB
{
"format": "orfb-design Graphormer inference checkpoints",
"format_version": 1,
"serialization": "PyTorch zip archive containing cfg and model",
"artifacts": [
{
"property": "reduction",
"filename": "graphormer_reduction.pt",
"sha256": "8893ba5b4de6dd541f59a070e517119f9828073f227bbe5082e25ea6d8e3bc8a",
"size_bytes": 193146717,
"source_filename": "reduction_checkpoint_best.pt",
"source_sha256": "978e7dbce43aaf614f7c9f6393bd293d5f6232f20ee6128284b55ba551a61290",
"state_dict_entries": 210,
"architecture": "graphormer_base",
"encoder_layers": 12,
"encoder_embed_dim": 768,
"encoder_attention_heads": 32
},
{
"property": "oxidation",
"filename": "graphormer_oxidation.pt",
"sha256": "2961e45c906a177c398aa61b9c3b2d28259e88e1f88aae787e77e8b72f21c8f5",
"size_bytes": 193149533,
"source_filename": "oxidation_checkpoint_best.pt",
"source_sha256": "d5c5cb62cf17217eadfb5b334e568ced000e0f6eb5871ce81bd62851a4853f09",
"state_dict_entries": 210,
"architecture": "graphormer_base",
"encoder_layers": 12,
"encoder_embed_dim": 768,
"encoder_attention_heads": 32
}
]
}