search

JinmiaoChenLab / GraphST

GNU Affero General Public License v3.0
92 stars 14 forks source link

specify batch size #34

Open jairaj-mathur opened 6 months ago

jairaj-mathur commented 6 months ago

Hi

I am working with a large dataset and model training gives me error that it cant allocate the required space on the GPU. Is there a way to specify batch size so that I can run the model training?

Warm regards

Jai

longyahui commented 6 months ago

Hi Jai,

Thank you for using our GraphST.

The GraphST model is not trained in batch. So we cannot specify batch size. But we would extend the model by revising the training way so that it can analyze more large dataset. Sorry for any inconvenience caused.

Best regards,
Yahui