search

PeterL1n / RobustVideoMatting

Robust Video Matting in PyTorch, TensorFlow, TensorFlow.js, ONNX, CoreML!
https://peterl1n.github.io/RobustVideoMatting/
GNU General Public License v3.0
8.53k stars 1.13k forks source link

CUDA run out of memory #128

Closed HarrytheOrange closed 2 years ago

HarrytheOrange commented 2 years ago

I run inference on 1060 6G, after several usages, the programme can take about 5Gb GPU memory after running but before running the inference. torch.cuda.empty_cache did not work.

PeterL1n commented 2 years ago

Set reduce seq_chunk or set seq_chunk=1 will help.

HarrytheOrange commented 2 years ago

thanks so much