custom-gcn / config.json
Huhujingjing's picture
Upload model
40fcca6
{
"architectures": [
"GCNModel"
],
"auto_map": {
"AutoConfig": "configuration_gcn.GCNConfig",
"AutoModel": "modeling_gcn.GCNModel"
},
"emb_input": 20,
"hidden_size": 64,
"input_feature": 64,
"model_type": "gcn",
"n_layers": 6,
"num_classes": 1,
"torch_dtype": "float32",
"transformers_version": "4.29.2"
}