DeepGraphLearning / GearNet

GearNet and Geometric Pretraining Methods for Protein Structure Representation Learning, ICLR'2023 (https://arxiv.org/abs/2203.06125)
MIT License
273 stars 27 forks source link

Working with batches of proteins with different sizes #66

Open mpedraza98 opened 4 months ago

mpedraza98 commented 4 months ago

Is there a way to work with proteins with different number of residues in a the same batch? I don't want to use the trimming transformation since I need the full protein, however, when using proteins of different lengths it always gives me an error when the batch size is different than 1.

I appreciate any suggestions on how to handle this issue