Out of memory

[TREASUREPPY]

Hello, the following problems occurred during the training. May I ask what caused them and what are the solutions?I'm using cuda10.1,python3.6.10.Waiting for your reply online.Thank you.
RuntimeError: cuda runtime error (2) : out of memory at /pytorch/torch/lib/THC/generic/THCStorage.cu:58

[XiaohangZhan]

Try to reduce the batch size.

[TREASUREPPY]

I changed the batch size, but this problem occurred again.RuntimeError: value cannot be converted to type double without overflow: inf-------- 原始邮件 --------发件人： Xiaohang Zhan <notifications@github.com>日期： 2020年5月12日周二 11:45收件人： XiaohangZhan/face_recognition_framework <face_recognition_framework@noreply.github.com>抄送： TREASUREPPY <PPY_1203@163.com>, Author <author@noreply.github.com>主 题： Re: [XiaohangZhan/face_recognition_framework] Out of memory (#6) Try to reduce the batch size. —You are receiving this because you authored the thread.Reply to this email directly, view it on GitHub, or unsubscribe.

[Linsongrong]

@XiaohangZhan hello,I also encountered the same problem, the problem of reducing batchsize to 1 still exists.