Training model with pyg graphs

[mrjoness]

Hello, I am using this model to train on protein data where each graph has a different number of atoms. So far I see good performance by padding shorter sequences, however I'd like to avoid this as I scale the model to much larger sequences. Is there currently any support for passing in batches of differently sized graphs (as in PyG)?

[dmitrii-khizbullin]

Hello, I am using this model to train on protein data where each graph has a different number of atoms. So far I see good performance by padding shorter sequences, however I'd like to avoid this as I scale the model to much larger sequences. Is there currently any support for passing in batches of differently sized graphs (as in PyG)?

I suppose you'd want to go with EGNN_Sparse.