Tested test_model at epoch 5
