Commit f5ae1a43 authored by lucas_miranda's avatar lucas_miranda
Browse files

Added extra branch to main autoencoder for rule_based prediction

parent f1f49bc1
Pipeline #99737 canceled with stages
in 4 minutes and 51 seconds
......@@ -83,7 +83,7 @@ def test_get_callbacks(
@settings(max_examples=10, deadline=None, suppress_health_check=[HealthCheck.too_slow])
@settings(max_examples=15, deadline=None, suppress_health_check=[HealthCheck.too_slow])
loss=st.one_of(st.just("ELBO"), st.just("MMD"), st.just("ELBO+MMD")),
next_sequence_prediction=st.one_of(st.just(0.0), st.just(1.0)),
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