New version with shuffle fix

This fixes a bug for small networks that would never finish shuffling the data in databricks and thus would never progress from training to detection