Test fails on Python 3.8 on CI

[JustinShenk]

Assertion in test_ae_classification_network_converges in test_models.py fails in CI on Python 3.8 for unknown reason.

>       assert classification_loss_post_training < classification_loss_pre_training
E       assert 0.04000916234824969 < 0.040007756348105446