Closed ilos-vigil closed 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].
--use-aim
use_aim
True
[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
Additional information
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 ofuse_aim
on Trainer class isTrue
[2].[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