specify batch size

[jairaj-mathur]

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]

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