Temperature reset after each epoch

YongfeiYan / Gumbel_Softmax_VAE

PyTorch implementation of a Variational Autoencoder with Gumbel-Softmax Distribution

203 stars 37 forks source link

Open Nintorac opened 6 years ago

Nintorac commented 6 years ago

Hey, sorry I haven't run your code to check, but here does this reset the temperature parameter after each epoch?

gokceneraslan commented 4 years ago

Seems like it resets in every 100 batch, so not after each epoch.