You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Sorry for the late reply but for some reason I stopped receiving notifications.
As the error suggests, you're trying to feed to the model a single graph with no leading batch dimension. You should reshape the features and adjacency matrix so that they have a batch size of 1.
Hello!
I have a custom dataset of graphs composed from invoice receipts (receipts vary in the number of words, so the number of nodes varies too).
I use GCN model for node-level prediction:
model = GCN(n_labels=2, channels=16, activation='relu', output_activation='softmax', dropout_rate=0)
Batch loader is used to train the model:
After the training the following prediction runs successfully:
model([features, adj])
But after saving the model the prediction fails:
With this error:
How can I save the model so that it works like just after the training?
The text was updated successfully, but these errors were encountered: