CNN distributed training with Spark and DL4J