PetarV- / GAT

Graph Attention Networks (https://arxiv.org/abs/1710.10903)
https://petar-v.com/GAT/
MIT License
3.15k stars 643 forks source link

How can I change the "batch_size" #38

Open dengzx7 opened 4 years ago

dengzx7 commented 4 years ago

I directly change the "batch_size" in file "execute_cora.py" from 1 to 10, and the following error occurred.

ValueError: Cannot feed value of shape (1, 2708, 1433) for Tensor 'input/Placeholder:0', which has shape '(10, 2708, 1433)'

So how can I change the "batch_size" when training or running the model ? Thx

liujian19911023 commented 4 years ago

I directly change the "batch_size" in file "execute_cora.py" from 1 to 10, and the following error occurred.

ValueError: Cannot feed value of shape (1, 2708, 1433) for Tensor 'input/Placeholder:0', which has shape '(10, 2708, 1433)'

So how can I change the "batch_size" when training or running the model ? Thx

change placeholder batch_size to None,but this is not useful,every epoch the model use full data, not node level。when graph is very big, need the change code to support node level train。

tanjia123456 commented 3 years ago

hello, do you know how to modify the value of the batch-size? It seems can't change it in the layer.py If you don't mind ,could you please share experience of it?

tanjia123456 commented 3 years ago

hello, do you know how to modify the value of the batch-size? It seems can't change it in the layer.py If you don't mind ,could you please share experience of it? This problem has been bothering me for a week, and I really can’t find a clue

wenwen19980927 commented 2 years ago

hi, do you know how to modify the batchsize now? ^^

hazdzz commented 2 years ago

GAT is a full batch GNN model. If you require a mini-batch model, you need to graph sampling first.