VAE中KL散度设计
之前训练结果
![](https://img.haomeiwen.com/i7768672/7a5fa033e8bdaacd.png)
KL散度消失
Generating Sentences from a Continuous Space
![](https://img.haomeiwen.com/i7768672/7c0931e24cb3cdb7.jpg)
![](https://img.haomeiwen.com/i7768672/a06fb4957c596747.jpg)
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'
![](https://img.haomeiwen.com/i7768672/3412f8e9a7b14475.png)
When
x0 = 2500
ann_function = 'linear'
![](https://img.haomeiwen.com/i7768672/2ec418c02ac3a586.png)
![](https://img.haomeiwen.com/i7768672/2b7273cedbc90559.png)
网友评论