search

yunjey / stargan

StarGAN - Official PyTorch Implementation (CVPR 2018)
MIT License
5.22k stars 968 forks source link

Error while running this code on same dataset #68

Open Rabia-Raja opened 6 years ago

Rabia-Raja commented 6 years ago

I am running this code. But I am facing some errors. Can you help me? The errors are: Traceback (most recent call last): File "main.py", line 110, in main(config) File "main.py", line 32, in main 'CelebA', config.mode, config.num_workers) File "/home/chenky/rabia/star-gan/StarGAN-master/data_loader.py", line 84, in get_loader dataset = CelebA(image_dir, attr_path, selected_attrs, transform, mode) File "/home/chenky/rabia/star-gan/StarGAN-master/data_loader.py", line 24, in init self.preprocess() File "/home/chenky/rabia/star-gan/StarGAN-master/data_loader.py", line 49, in preprocess idx = self.attr2idx[attr_name] KeyError: 'Black_Hair

FighterFong commented 5 years ago

Are you fix it?

iamzhangyunlu commented 4 years ago

I am running this code. But I am facing some errors. Can you help me? The errors are: �Traceback (most recent call last): File "main.py", line 110, in main(config) File "main.py", line 32, in main 'CelebA', config.mode, config.num_workers) File "/home/chenky/rabia/star-gan/StarGAN-master/data_loader.py", line 84, in get_loader dataset = CelebA(image_dir, attr_path, selected_attrs, transform, mode) File "/home/chenky/rabia/star-gan/StarGAN-master/data_loader.py", line 24, in init self.preprocess() File "/home/chenky/rabia/star-gan/StarGAN-master/data_loader.py", line 49, in preprocess idx = self.attr2idx[attr_name] KeyError: 'Black_Hair

Same problem here! Have you fixed yet please? @Rabia-Raja