gradient check passed but has numerial issues in high dimensions