Where batch_size has been defined?

[dunghoilyd01]

I need to run code on on CUDA, but even colab doesn't have enough VRAM to run it. So i am trying to decrease the batch_size, but dont know where to modify it. Can you tell me where is it defined?

[Ewillingfly]

As far as i know, there are't samples in this work. You see, the entire dataset is a single graph.

[lizhixi212]

torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 3.99 GiB (GPU 1; 23.70 GiB total capacity; 17.64 GiB already allocated; 1.24 GiB free; 21.34 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF