/stylegan2-deepspeed

Implementation of stylegan2 using deepspeed

Primary LanguagePythonMIT LicenseMIT

stylegan2-deepspeed

An implementation of stylegan2 using Microsoft's DeepSpeed.

Currently WIP, lots of things unimplemented.

GAN implemenation lifted from lucidrain's repo.

Example Commandline

deepspeed `which stylegan2_deepspeed` --deepspeed --deepspeed_config ./config/stylegan2.json -d /path/to/input/images

References

TODO