fixed up training parallelism to actually be parallel