-
Notifications
You must be signed in to change notification settings - Fork 126
/
run_bedroom128.py
27 lines (23 loc) · 767 Bytes
/
run_bedroom128.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
from templates import *
from templates_latent import *
if __name__ == '__main__':
# train the autoenc moodel
# this requires V100s.
gpus = [0, 1, 2, 3]
conf = bedroom128_autoenc()
train(conf, gpus=gpus)
# infer the latents for training the latent DPM
# NOTE: not gpu heavy, but more gpus can be of use!
gpus = [0, 1, 2, 3]
conf.eval_programs = ['infer']
train(conf, gpus=gpus, mode='eval')
# train the latent DPM
# NOTE: only need a single gpu
gpus = [0]
conf = bedroom128_autoenc_latent()
train(conf, gpus=gpus)
# unconditional sampling score
# NOTE: a lot of gpus can speed up this process
gpus = [0, 1, 2, 3]
conf.eval_programs = ['fid(10,10)']
train(conf, gpus=gpus, mode='eval')