Letting user generate sample/show progress without installing Aim - Githubissues

lucidrains / lightweight-gan

Implementation of 'lightweight' GAN, proposed in ICLR 2021, in Pytorch. High resolution image generations that can be trained within a day or two

MIT License

1.62k stars 220 forks source link

Letting user generate sample/show progress without installing Aim #137

Closed ilos-vigil closed 2 years ago

ilos-vigil commented 2 years ago

Currently user who don't install Aim unable to generate sample or show progress without installing Aim. It's due to "model_args" doesn't carry value of args --use-aim[1] and default value of use_aim on Trainer class is True[2]. I considered modifying dict "model_args" instead[3], but it cause error on different Trainer class initialization.

[1] https://github.com/lucidrains/lightweight-gan/blob/main/lightweight_gan/cli.py#L126 [2] https://github.com/lucidrains/lightweight-gan/blob/main/lightweight_gan/lightweight_gan.py#L983 [3] https://github.com/lucidrains/lightweight-gan/pull/136