Flux.train! runs through 1 epoch of training, that is it goes through the training data once. The matlab example linked uses about 999,999 more iterations, so I think you’d have more success by training for a little longer 
1 Like