XingangPan / DragGAN

Official Code for DragGAN (SIGGRAPH 2023)
https://vcai.mpi-inf.mpg.de/projects/DragGAN/
Other
35.65k stars 3.44k forks source link

Loading Error on windows 11 #390

Open Azum91 opened 1 year ago

Azum91 commented 1 year ago

`Microsoft Windows [Version 10.0.22621.2134] (c) Microsoft Corporation. All rights reserved.

I:\GAN work\DragGAN>I:\my_python_env\Scripts\activate

(my_python_env) I:\GAN work\DragGAN>sh scripts/gui.sh Loading "I:\GAN work\DragGAN\checkpoints\stylegan2_lions_512_pytorch.pkl"... Done. () {'z_dim': 512, 'c_dim': 0, 'w_dim': 512, 'img_resolution': 512, 'img_channels': 3, 'mapping_kwargs': {'num_layers': 8, 'embed_features': None, 'layer_features': None, 'activation': 'lrelu', 'lr_multiplier': 0.01, 'w_avg_beta': 0.995}, 'synthesis_kwargs': {'channel_base': 32768, 'channel_max': 512, 'num_fp16_res': 0, 'conv_clamp': None, 'architecture': 'skip', 'resample_filter': [1, 3, 3, 1], 'use_noise': True, 'activation': 'lrelu'}} Setting up PyTorch plugin "bias_act_plugin"... `

It seems that the model gets stuck on loading. It may be because of GPU or something else.

System Specs:

  1. CPU i7 9th Gen 12 Logical Core
  2. 16 GB GDDR4 RAM
  3. 6 GB GPU GTX 1660Ti
  4. 1 TB WD Nvme

In order to download the models I had to modify scripts/download_model.sh for Windows, and fixed a few errors that were occurring due to missing plugins and libraries. After that, the model doesn't load.