Commit 2df67bb6 authored by lucas_miranda's avatar lucas_miranda
Browse files

Started implementing annealing mode for KL divergence

parent 5b8b49c2
Pipeline #100117 passed with stages
in 19 minutes and 20 seconds
......@@ -183,7 +183,10 @@ def test_KLDivergenceLayer(annealing_mode):
weight=1.0,
)(x)
kl_deepof = deepof.model_utils.KLDivergenceLayer(
distribution_b=prior, iters=1, warm_up_iters=0, annealing_mode=annealing_mode,
distribution_b=prior,
iters=1,
warm_up_iters=0,
annealing_mode=annealing_mode,
)(x)
test_model = tf.keras.Model(i, [kl_canon, kl_deepof])
......@@ -231,7 +234,7 @@ def test_MMDiscrepancyLayer(annealing_mode):
prior=prior,
iters=1,
warm_up_iters=0,
annealing_mode=annealing_mode
annealing_mode=annealing_mode,
)(x)
test_model = tf.keras.Model(i, x)
......
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