search

TencentARC / VQFR

ECCV 2022, Oral, VQFR: Blind Face Restoration with Vector-Quantized Dictionary and Parallel Decoder
https://ycgu.site/projects/vqfr
Other
324 stars 42 forks source link

4090 not worked... only CPU mode? #22

Open ffdown opened 1 year ago

ffdown commented 1 year ago 
(vqfr) E:\AI\VQFR>python demo.py -i inputs/whole_imgs -o results -v 2.0 -f 0.1
demo.py:69: UserWarning: The unoptimized RealESRGAN is slow on CPU. We do not use it. If you really want to use it, please modify the corresponding codes.
  warnings.warn('The unoptimized RealESRGAN is slow on CPU. We do not use it. '
e:\Anaconda3\envs\vqfr\lib\site-packages\torchvision\models\_utils.py:208: UserWarning: The parameter 'pretrained' is deprecated since 0.13 and may be removed in the future, please use 'weights' instead.
  warnings.warn(
e:\Anaconda3\envs\vqfr\lib\site-packages\torchvision\models\_utils.py:223: UserWarning: Arguments other than a weight enum or `None` for 'weights' are deprecated since 0.13 and may be removed in the future. The current behavior is equivalent to passing `weights=None`.
  warnings.warn(msg)

Bug, or not? How i can start this script with cuda on 4090?? GPU load 0!

image