from templates import * | |
from templates_latent import * | |
if __name__ == '__main__': | |
gpus = [0, 1, 2, 3] | |
conf = bedroom128_ddpm() | |
train(conf, gpus=gpus) | |
gpus = [0, 1, 2, 3] | |
conf.eval_programs = ['fid10'] | |
train(conf, gpus=gpus, mode='eval') |
from templates import * | |
from templates_latent import * | |
if __name__ == '__main__': | |
gpus = [0, 1, 2, 3] | |
conf = bedroom128_ddpm() | |
train(conf, gpus=gpus) | |
gpus = [0, 1, 2, 3] | |
conf.eval_programs = ['fid10'] | |
train(conf, gpus=gpus, mode='eval') |