search

genforce / mganprior

[CVPR 2020] Image Processing Using Multi-Code GAN Prior
https://genforce.github.io/mganprior/
289 stars 44 forks source link

GPU usage goes up by 2mb every few seconds #24

Closed gzhhhere closed 2 years ago

gzhhhere commented 2 years ago

CUDA_VISIBLE_DEVICES=2 python multi_code_inversion.py \ --gan_model=pggan_celebahq \ --target_images=/mnt/DataDrive164/lirj/JGYYDS/celeba_mask_hq_train_test/test/img \ --outputs=/mnt/DataDrive164/lirj/JGYYDS/mganprior_pggan/celeba_test1000_out1024_multi20_compose8 \ --composing_layer=8 \ --z_number=20 \ --video=False \

Here is my command, Can this problem be solved?