search

mrlibw / ControlGAN

Pytorch implementation for Controllable Text-to-Image Generation.
165 stars 37 forks source link

Issue while running train command #10

Open ShivamGuptaHat opened 2 years ago

ShivamGuptaHat commented 2 years ago

Dear Sir, I am shivam Gupta an MTech student. I have chosen control-GAN as my thesis topic and I found your paper and this project really interesting. In the recent few days, I have been trying to run and understand your code. But I am getting one error. I tried a lot to resolve it. But it is not working. Can you please have a look at it?

After following all the steps, I am trying to run the below command-

python main.py --cfg cfg/train_bird.yml --gpu 0

I am running this project on google colab.

Here I am sharing an image that contains that error... error image

LINGQINGZHU commented 2 years ago

Hi, I recently encountered the same problem as you. I compared the previous code and found the error. You can add a parameter to pyramid_expand() on line 129 in utils.py: multichannel=True image