cudnn.benchmark = False