Why is the Batch Size for Downstream Tasks Set to 2? It Seems Quite Slow

DeepGraphLearning / GearNet

GearNet and Geometric Pretraining Methods for Protein Structure Representation Learning, ICLR'2023 (https://arxiv.org/abs/2203.06125)

MIT License

263 stars 27 forks source link

Why is the Batch Size for Downstream Tasks Set to 2? It Seems Quite Slow #65

Open minjiecheng opened 2 months ago

minjiecheng commented 2 months ago

Why is the batch size for downstream tasks set to 2? I can run it with a batch size of 128 or more, and setting it to 2 makes the running speed too slow.