File size: 264 Bytes
1ab03a3
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
from templates import *
from templates_latent import *

if __name__ == '__main__':
    gpus = [0, 1, 2, 3]
    conf = ffhq128_ddpm_130M()
    train(conf, gpus=gpus)

    gpus = [0, 1, 2, 3]
    conf.eval_programs = ['fid10']
    train(conf, gpus=gpus, mode='eval')