search

Maclory / Deep-Iterative-Collaboration

Pytorch implementation of Deep Face Super-Resolution with Iterative Collaboration between Attentive Recovery and Landmark Estimation (CVPR 2020)
MIT License
295 stars 63 forks source link

while running test.py(RuntimeError: CUDA out of memory) #20

Closed HacxS closed 4 years ago

HacxS commented 4 years ago

image

Help Please

0%| | 0/1 [00:00<?, ?it/s] Traceback (most recent call last): File "test.py", line 128, in main() File "test.py", line 69, in main solver.test() File "/home/govind/audio-video/image-enhance/Deep-Iterative-Collaboration/code/solvers/SRLandmarkGANSolver.py", line 220, in test SR_list, heatmap_list = forward_func(self.LR) File "/home/govind/anaconda3/lib/python3.7/site-packages/torch/nn/parallel/data_parallel.py", line 150, in forward return self.module(*inputs[0], kwargs[0]) File "/home/govind/anaconda3/lib/python3.7/site-packages/torch/nn/modules/module.py", line 532, in call result = self.forward(*input, *kwargs) File "/home/govind/audio-video/image-enhance/Deep-Iterative-Collaboration/code/networks/dic_arch.py", line 89, in forward FB_out_first = self.first_block(x) File "/home/govind/anaconda3/lib/python3.7/site-packages/torch/nn/modules/module.py", line 532, in call result = self.forward(input, kwargs) File "/home/govind/audio-video/image-enhance/Deep-Iterative-Collaboration/code/networks/srfbn_hg_arch.py", line 130, in forward LD_H = self.downtranBlocksidx-1 File "/home/govind/anaconda3/lib/python3.7/site-packages/torch/nn/modules/module.py", line 532, in call result = self.forward(*input, *kwargs) File "/home/govind/anaconda3/lib/python3.7/site-packages/torch/nn/modules/container.py", line 100, in forward input = module(input) File "/home/govind/anaconda3/lib/python3.7/site-packages/torch/nn/modules/module.py", line 532, in call result = self.forward(input, **kwargs) File "/home/govind/anaconda3/lib/python3.7/site-packages/torch/nn/modules/conv.py", line 345, in forward return self.conv2d_forward(input, self.weight) File "/home/govind/anaconda3/lib/python3.7/site-packages/torch/nn/modules/conv.py", line 342, in conv2d_forward self.padding, self.dilation, self.groups) RuntimeError: CUDA out of memory. Tried to allocate 3.00 GiB (GPU 0; 15.90 GiB total capacity; 12.32 GiB already allocated; 2.86 GiB free; 12.34 GiB reserved in total by PyTorch)

and current configurations: { "name": "CelebA_GAN", "mode": "sr_align_gan", "degradation": "BI", "gpu_ids": [0], "use_tb_logger": false, "scale": 8, "is_train": false, "rgb_range": 1, "save_image": true, "datasets": { "test_CelebA": { "mode": "LRHR", "name": "CelebA", "dataroot_HR": "/home/govind/audio-video/image-enhance/Deep-Iterative-Collaboration/image/", "dataroot_LR": "/home/govind/audio-video/image-enhance/Deep-Iterative-Collaboration/image/", single image inside "data_type": "img", "LR_size": 16, "HR_size": 128 } }, "networks": { "which_model": "DIC", "num_features": 48, "in_channels": 3, "out_channels": 3, "num_steps": 4, "num_groups": 6, "detach_attention": false, "hg_num_feature": 256, "hg_num_keypoints": 68, "num_fusion_block": 7 }, "solver": { "pretrained_path": "/home/govind/audio-video/image-enhance/Deep-Iterative-Collaboration/models/DICGAN_CelebA.pth" }, "path": { "root": "../" } }

forMUmu commented 3 years ago

How to do? I have got this problem. I don't know how to solve it.

malinjie-hub commented 2 years ago

I have got this problem too，Hope to get help