Fix? error in new parallel cost function. Was dividing cost by input batch size instead of total input size.