mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Notes for lucidrains converter.

Browse Source
This commit is contained in:
James Betker 2020-12-18 09:55:38 -07:00
parent b4720ea377
commit c717765bcb
2 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
codes/models/stylegan/stylegan2_lucidrains_conformed.py
View File

@ -1,3 +1,7 @@
# Note this is an attempt to conform the lucidrains stylegan implementation to the official reference spec so that
# I could use pretrained weights from NVIDIA. It is not currently successful, but that may be due to the weight
# converter and not the code changes here. Use at your own risk.
import math import math
import multiprocessing import multiprocessing
from contextlib import contextmanager, ExitStack from contextlib import contextmanager, ExitStack

6
codes/scripts/stylegan2/convert_weights_lucidrains.py
View File

@ -4,6 +4,12 @@
# #
# Also doesn't require you to install Tensorflow 1.15 or clone the nVidia repo. # Also doesn't require you to install Tensorflow 1.15 or clone the nVidia repo.
# THIS DOES NOT CURRENTLY WORK.
# It does transfer all weights from the stylegan model to the lucidrains one, but does not produce correct results.
# The rosinality script this was stolen from has some "odd" intracacies that may be at cause for this: for example
# weight "flipping" in the conv layers which I do not understand. It may also be because I botched some of the mods
# required to make the lucidrains implementation conformant. I'll (maybe) get back to this some day.
import argparse import argparse
import os import os
import sys import sys