Commit f7468519 authored by lucas_miranda's avatar lucas_miranda
Browse files

Implemented 1 cycle learning rate schedule

parent c1d1d84f
......@@ -195,7 +195,7 @@ class SEQ_2_SEQ_GMVAE:
if self.prior == "standard_normal":
init_means = far_away_uniform_initialiser(
[self.number_of_components, self.ENCODING], minval=0, maxval=15
[self.number_of_components, self.ENCODING], minval=0, maxval=5
)
self.prior = tfd.mixture.Mixture(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment