DualStyleGAN/model/stylegan/op_cpu at main · williamyang1991/DualStyleGAN · GitHub

Name		Name	Last commit message	Last commit date
parent directory ..
__init__.py		__init__.py
conv2d_gradfix.py		conv2d_gradfix.py
fused_act.py		fused_act.py
readme.md		readme.md
upfirdn2d.py		upfirdn2d.py

readme.md

Code from rosinality-stylegan2-pytorch-cp

Scripts to convert rosinality/stylegan2-pytorch to the CPU compatible format

If you would like to use CPU for testing, please make the following changes:

Change model.stylegan.op to model.stylegan.op_cpu

DualStyleGAN/util.py

Line 10 in 6ab383b

from model.stylegan.op import conv2d_gradfix
Change model.stylegan.op to model.stylegan.op_cpu

DualStyleGAN/model/stylegan/model.py

Line 11 in 6ab383b

from model.stylegan.op import FusedLeakyReLU, fused_leaky_relu, upfirdn2d, conv2d_gradfix

If using GPUs, changing the above lines back to the original ones.