File size: 396 Bytes
f57bf3e |
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]
|