non_blocking参数的设置。
问题:images.cuda(non_blocking=True),target.cuda(non_blocking=True)把数据迁移到GPU里面;nn.DataParallel(model.to(device), device_ids=gpus, output_device=gpus[0])包装模型。images.cuda(non_blocking=True)为何要设置参数non_blocking=True呢?解释:non_blocking默认值为False, 通常我们会在加载数据时,将D