Okay, wasn't really accumulating all training cases in one bit matrix. Fixed that and now GPU runs out of resources. Need to split computation into smaller mats.