BioinfoMachineLearning / DeepInteract

A geometric deep learning pipeline for predicting protein interface contacts. (ICLR 2022)
https://zenodo.org/record/6671582
GNU General Public License v3.0
62 stars 11 forks source link

About the training resources #11

Closed terry-r123 closed 2 years ago

terry-r123 commented 2 years ago

Hi @amorehead, I would like to know how long do you need to train on 72 v100 GPUs? And is it 30 or 50 epochs?

Thanks!

amorehead commented 2 years ago

@terry-r123, with 72 v100 GPUs, DeepInteract models took around 2 hours to train for approximately 40 epochs each.

terry-r123 commented 2 years ago

@amorehead, Thanks very much!

terry-r123 commented 2 years ago

Hi @amorehead, What is the batch size with 72 v100 GPUs for around 2 hours?

amorehead commented 2 years ago

With this default configuration, the effective batch size is 72.

terry-r123 commented 2 years ago

With this default configuration, the effective batch size is 72.

Thanks! I get it!