Hi, the error message looks like the following:
RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 10.92 GiB total capacity; 10.34 GiB already allocated; 9.38 MiB free; 10.34 GiB reserved in total by PyTorch)
It occured when executing the following code:
for i_batch, sample_batched in enumerate(TraindataLoader):
local_X = sample_batched[‘input’].type(dtype)
local_Y = sample_batched[‘EEGResponse’].type(dtype)
yhat = model(local_X)     -----error occurred
looking forward to your suggestion.