CUDA out of memory error

tencent-ailab / pika

a lightweight speech processing toolkit based on Pytorch and (Py)Kaldi

Apache License 2.0

339 stars 57 forks source link

Open boji123 opened 3 years ago

boji123 commented 3 years ago

I set batchsize=1 layer=1 but it still OOM

cweng6 commented 3 years ago

what GPU you were using? and how large is the GPU memory