stylegan2 환경 구축

Data-science/deep learning

study&grow 2020. 10. 28. 04:55

728x90

requirements에는 cuda 10.0으로 나와있는데

cuda 10.1로 해야 된다.

pytorch repo

rosinality/stylegan2-pytorch

Implementation of Analyzing and Improving the Image Quality of StyleGAN (StyleGAN 2) in PyTorch - rosinality/stylegan2-pytorch

github.com

tensorflow 공식 버전보다 훨씬 좋다고 느껴짐...