RuntimeError: CUDA out of memory about recoder HOT 9 OPEN

Hi @pkuzqh , I've got another issue when running the code.

from recoder.

If you want to change the batch size, you need to change the number in the dict "args". If you want to use multiple GPUs, you need to modify "model = nn.DataParallel(model, device_ids=[0, 1])".

from recoder.

Hi @pkuzqh, thank you for the reply. The cuda out of memory issue has been resolved. However, I found the new error above. Please kindly suggest, thanks.

from recoder.

How many GPUs do you use? And the batch size?

from recoder.

3, I indicated in the train() that: device_ids=[1,2,3]
the batch size is 16

from recoder.

You need to change the number "4" in line 103-106 to a multiple of 3. And the batch size needs to be a multiple of 3.

from recoder.

OK, thank you very much @pkuzqh ! It can now run. However, I saw in the train(), the number of epochs is 100000, for epoch in range(100000): is that true?

from recoder.

BTW, for inference, it looks like the testDefect4j.py can only use 1 GPU? Since I have 4 GPUs, only one was used, and it caused an OOM issue.

from recoder.

you can use "nn.DataParallel" to use multiple gpus in testDefect4J.py

from recoder.