Skip to content

Latest commit

 

History

History
14 lines (8 loc) · 682 Bytes

readme.md

File metadata and controls

14 lines (8 loc) · 682 Bytes

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:

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

    from model.stylegan.op import conv2d_gradfix

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

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

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