Out of Memory

[kdrkdrkdr]

I used ESRGAN Mode. But I can't use this..

CUDA-compatible GPU located!
Starting ESRGAN detection and decensor
Model warmup complete
Exception in Tkinter callback
Traceback (most recent call last):
File "C:\Anaconda3\envs\deeplearning\lib\tkinter_init_.py", line 1705, in call
return self.func(*args)
File "main.py", line 311, in
go_button = Button(mos_win, text="Go!", command = lambda: hentAI_TGAN(in_path=o_entry.get(), is_video=False))
File "main.py", line 184, in hentAI_TGAN
detect_instance.run_ESRGAN(in_path = in_path, is_video = is_video, force_jpg = force_jpg)
File "C:\Users\power\Desktop\Project\Dev\GoodbyeForeignLangAndCensored\hent-AI\detector.py", line 379, in run_ESRGAN
self.resize_GAN(img_path=img_path, img_name=img_name, is_video=is_video)
File "C:\Users\power\Desktop\Project\Dev\GoodbyeForeignLangAndCensored\hent-AI\detector.py", line 220, in resize_GAN
self.esrgan_instance.run_esrgan(test_img_folder=file_name, out_filename=gan_img_path, mosaic_res=granularity)
File "C:\Users\power\Desktop\Project\Dev\GoodbyeForeignLangAndCensored\hent-AI\ColabESRGAN\test.py", line 49, in run_esrgan
output = self.model(img_LR).data.squeeze().float().cpu().clamp_(0, 1).numpy()
File "C:\Anaconda3\envs\deeplearning\lib\site-packages\torch\nn\modules\module.py", line 477, in call
result = self.forward(*input, **kwargs)
File "C:\Users\power\Desktop\Project\Dev\GoodbyeForeignLangAndCensored\hent-AI\ColabESRGAN\block.py", line 83, in forward
output = x + self.sub(x)
File "C:\Anaconda3\envs\deeplearning\lib\site-packages\torch\nn\modules\module.py", line 477, in call
result = self.forward(*input, **kwargs)
File "C:\Anaconda3\envs\deeplearning\lib\site-packages\torch\nn\modules\container.py", line 91, in forward
input = module(input)
File "C:\Anaconda3\envs\deeplearning\lib\site-packages\torch\nn\modules\module.py", line 477, in call
result = self.forward(*input, **kwargs)
File "C:\Users\power\Desktop\Project\Dev\GoodbyeForeignLangAndCensored\hent-AI\ColabESRGAN\block.py", line 229, in forward
out = self.RDB3(out)
File "C:\Anaconda3\envs\deeplearning\lib\site-packages\torch\nn\modules\module.py", line 477, in call
result = self.forward(*input, **kwargs)
File "C:\Users\power\Desktop\Project\Dev\GoodbyeForeignLangAndCensored\hent-AI\ColabESRGAN\block.py", line 207, in forward
x5 = self.conv5(torch.cat((x, x1, x2, x3, x4), 1))
RuntimeError: CUDA error: out of memory