search

yzxing87 / pytorch-deep-video-prior

[NeurIPS 2020] Blind Video Temporal Consistency via Deep Video Prior
118 stars 22 forks source link

attempt to get argmax of an empty sequence #13

Open semel1 opened 2 years ago

semel1 commented 2 years ago

Upon running "main_IRT.py", got an error:

Exception has occurred: ValueError
attempt to get argmax of an empty sequence
  File "D:\colorization\deep-video-prior-master-pytorch\main_IRT.py", line 49, in <module>
    for x in subprocess.Popen("nvidia-smi -q -d Memory | grep -A4 GPU | grep Free", shell=True, stdout=subprocess.PIPE).stdout.readlines()]))

Happened in both Windows 10 and WSL2 Nvidia RTX 3090

yzxing87 commented 2 years ago

Seems something went wrong with your GPU driver. Is 'nvidia-smi' normal on your machine? Anyway, you can comment out line 49, and replace it with os.environ["CUDA_VISIBLE_DEVICES"]="0", for simplicity.

semel1 commented 2 years ago

Here report from nviia-sm on Windows 10 and WSL2i:

nvidia_smi

Anyway comment out line 49 only and replace it with os.environ["CUDA_VISIBLE_DEVICES"]="0", throws an error:

  File "main_IRT.py", line 50
    os.environ["CUDA_VISIBLE_DEVICES"] = '0'
     ^
SyntaxError: invalid syntax

However comment out both line 48 and 49 and replace them with os.environ["CUDA_VISIBLE_DEVICES"]="0" - did the job. Thank you for your prompt responce