train_pie_celeba_april.config

dataset = celeba
algorithm = pie
modelname = april

modellatentdim = 512
outerlayers = 20
innerlayers = 8
levels = 4
linlayers = 2
linchannelfactor = 2
lineartransform = lu
splinerange = 10.0
splinebins = 11
actnorm = True
batchnorm = False
dropout = 0.0
pieepsilon = 0.1

epochs = 500
batchsize = 25
lr = 3.0e-4
nllfactor = 1.0
uvl2reg = 0.01
weightdecay = 1.0e-5
clip = 5.0
validationsplit = 0.1

debug = True
dir = /sdcc/u/brehmer/manifold-flow