File size: 333 Bytes
1ab03a3 |
1 2 3 4 5 6 7 8 9 10 11 12 |
from templates_cls import *
from experiment_classifier import *
if __name__ == '__main__':
# need to first train the diffae autoencoding model & infer the latents
# this requires only a single GPU.
gpus = [0]
conf = ffhq256_autoenc_cls()
train_cls(conf, gpus=gpus)
# after this you can do the manipulation!
|