search

facebookresearch / denoiser

Real Time Speech Enhancement in the Waveform Domain (Interspeech 2020)We provide a PyTorch implementation of the paper Real Time Speech Enhancement in the Waveform Domain. In which, we present a causal speech enhancement model working on the raw waveform that runs in real-time on a laptop CPU. The proposed model is based on an encoder-decoder architecture with skip-connections. It is optimized on both time and frequency domains, using multiple loss functions. Empirical evidence shows that it is capable of removing various kinds of background noise including stationary and non-stationary noises, as well as room reverb. Additionally, we suggest a set of data augmentation techniques applied directly on the raw waveform which further improve model performance and its generalization abilities.
Other
1.62k stars 299 forks source link

CUDA out of memory #135

Open Ammara-Ihsan opened 1 year ago

Ammara-Ihsan commented 1 year ago

Hi, I'm new at this platform, can anyone pls help me how to solve this error, I tried to reduce epochs and batch_size, and I also tried to reduce the number of workers but nothing helped in this, Thanks,

RuntimeError: CUDA out of memory. Tried to allocate 752.00 MiB (GPU 0; 14.76 GiB total capacity; 9.84 GiB already allocated; 53.94 MiB free; 10.03 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF

cgvalle commented 1 year ago

Seems like your v-ram is been use for another process (9.84 GiB used something else). Use nvidia-smi to check which process are using the gpu