Setting epochs=1 to speed up tests.