diffae / run_bedroom128_ddim.py
qninhdt's picture
Upload 68 files
1ab03a3 verified
raw
history blame contribute delete
262 Bytes
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')