File size: 396 Bytes
fc1b0fd
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
#!/usr/bin/env python3

import torch
import pickle

with open('../models/gamma500/network-snapshot-010000.pkl', 'rb') as f:
    G = pickle.load(f)['G_ema']# torch.nn.Module
z = torch.randn([1, G.z_dim])# latent codes
c = None                                # class labels (not used in this example)
img = G(z, c, force_fp32=True)                           # NCHW, float32, dynamic range [-1, +1]