| { | |
| "architectures": [ | |
| "MoleBERTTokenizer" | |
| ], | |
| "atom_vocab_size": 1024, | |
| "gnn_encoder_dropout": 0.1, | |
| "gnn_encoder_embedding_dim": 300, | |
| "gnn_encoder_jk": "last", | |
| "gnn_encoder_layer_hidden_dim": 600, | |
| "gnn_encoder_num_layers": 5, | |
| "graph_reconstruct_dropout": 0.1, | |
| "graph_reconstruct_hidden_dim": 600, | |
| "re_build_edge": true, | |
| "torch_dtype": "float32", | |
| "transformers_version": "4.46.3", | |
| "vq_commitment_cost": 0.25 | |
| } | |