VAE中KL散度设计
之前训练结果

KL散度消失
Generating Sentences from a Continuous Space


def kl_anneal_function(ann_function, step, k, x0):
if ann_function == 'logistic':
return float(1/(1+np.exp(-k*(step-x0))))
elif ann_function == 'linear':
return min(1, step/x0)
When
k = 0.0025
x0 = 2500
ann_function = 'logistic'

When
x0 = 2500
ann_function = 'linear'


网友评论