search

Zongwei97 / SCTNet

[ICCV 23] Official implementation of Alignment-free HDR Deghosting with Sematnics Consistent Transformer (Code & Dataset)
23 stars 3 forks source link

CUDA out of Memory #4

Open maopengcheng924 opened 2 weeks ago

maopengcheng924 commented 2 weeks ago

I use 2 NVIDIA A100 80GB PCIe, 160GB GPU memory total, but I still got CUDA out of memory error. I didnot change ur code, just git clone, pip install and then python train. So I am confused, could u please tell me something about ur experiment env?

josair21 commented 2 weeks ago

I use Python 3.11.9 with 2 x A100 40GB without any problems. The curious thing is that I get lower results:

0: Average PSNR_mu: 43.8158 PSNR_l: 41.5196 0: Average SSIM_mu: 0.9917 SSIM_l: 0.9884

requirements.txt