search

hkchengrex / Mask-Propagation

[CVPR 2021] MiVOS - Mask Propagation module. Reproduced STM (and better) with training code :star2:. Semi-supervised video object segmentation evaluation.
https://hkchengrex.github.io/MiVOS/
MIT License
128 stars 22 forks source link

RuntimeError: CUDA error: out of memory #11

Closed longmalongma closed 3 years ago

longmalongma commented 3 years ago

How many GPUs do you need to test on Davis and YouTube?I keep reporting memory errors during my tests.I directly used the model trained by static pictures for VOS training, skipping the pre-training of BL30K. Is that OK?

hkchengrex commented 3 years ago

Test: One 11GB GPU. Yes.

longmalongma commented 3 years ago

Test: One 11GB GPU. Yes.

Test: One 11GB GPU. Yes.

Did an 11GB GPU seem inadequate during testing?Can you specify what block of GPU it is?Or can we test with multiple GPUs?

image image

hkchengrex commented 3 years ago

You are probably doing something wrong. The memory is full on all four GPUs already!

longmalongma commented 3 years ago

Test: One 11GB GPU. Yes.

@hkchengrex This ok? I've tried this and I'm still out of memory

CUDA_VISIBLE_DEVICES=0,1 python eval_davis.py

You are probably doing something wrong. The memory is full on all four GPUs already!

I got it!