editGAN_release icon indicating copy to clipboard operation
editGAN_release copied to clipboard

Published 20 hours ago • verified publisher icon nv-tlabs

how to convert the tensorflow checkpoint to pytorch?

Open 506143026 opened this issue 2 years ago • 3 comments

i want to use the 256x256 pre-trained stylegan2 model. there are some problems to convert the tensorflow checkpoint to pytorch. can you provide a convert demo?

506143026 avatar Jun 02 '22 04:06 506143026

Meet same problem!

haoyuhsu avatar Jun 27 '22 09:06 haoyuhsu

mee too

Ley-lele avatar Jul 22 '22 02:07 Ley-lele

Hi. stylegan2-pytorch codebase from rosinality gives a nice example! https://github.com/rosinality/stylegan2-pytorch/blob/master/convert_weight.py

arieling avatar Jul 22 '22 04:07 arieling