search

nashory / pggan-pytorch

:fire::fire: PyTorch implementation of "Progressive growing of GANs (PGGAN)" :fire::fire:
MIT License
819 stars 134 forks source link

I met come errors in trainer.py #46

Open ChenXiao61 opened 5 years ago

ChenXiao61 commented 5 years ago

Thanks for sharing your code for us ! I have trained for more than 2500 epochs on one GTX 1080 ti, but the resolution is still the init phase :4X4 and doesn't grow . And I met the error after 2557 epochs like this: AttributeError: 'DataParallel' object has no attribute 'grow_network'. I don't know how to solve.

kleinyoni commented 5 years ago

add module like this G.module.grow_network

This worked for me

ChenXiao61 commented 5 years ago

Thanks. @kleinyoni Where did I add module "G.module.grow_network"?

n1dhi commented 5 years ago

Change G.grow_network to G.module.grow_network and do the same thing for D.grow_network as well

ChenXiao61 commented 5 years ago

@n1dhi Thanks, did you mean change to this self.G.module.grow_network(floor(self.resl)) self.D.module.grow_network(floor(self.resl)) in the trainer.py?

txdon commented 4 years ago

@n1dhi Thanks, did you mean change to this self.G.module.grow_network(floor(self.resl)) self.D.module.grow_network(floor(self.resl)) in the trainer.py?

Hi, i met same error, could you solve this problem? please help me! thank you!