У меня есть вопрос, я скопировал код из Style-GAN в блокнот jupyter, но не работает должным образом, это код, который не работает:
# Initialize TensorFlow.
tflib.init_tf()
# Load pre-trained network.
#url = 'https://drive.google.com/uc?id=1MEGjdvVpUsu1jB4zrXZN7Y4kBBOzizDQ' # karras2019stylegan-ffhq-1024x1024.pkl
#with dnnlib.util.open_url(url, cache_dir=config.cache_dir) as f:
f="./karras2019stylegan-ffhq-1024x1024.pkl"
with open(f, 'rb') as pickle_file:
_G, _D, Gs = pickle.load(pickle_file)
# _G = Instantaneous snapshot of the generator. Mainly useful for resuming a previous training run.
# _D = Instantaneous snapshot of the discriminator. Mainly useful for resuming a previous training run.
# Gs = Long-term average of the generator. Yields higher-quality results than the instantaneous snapshot.
Проблема в
_G, _D, Gs = pickle.load(pickle_file)
У меня такая ошибка: TypeError: cannot unpack non-iterable int object
Я не знаю, почему не работает, потому что я использую тот же код, но в .py и его работе.
Кто-нибудь знает, почему не работает?
это репо: https://github.com/NVlabs/stylegan