1.model.cuda(async=True)
改为:model.cuda(non_blocking=True) 。python版本 python>=3.7
2.
total_loss += loss_val.data[0]
IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number
pytorch版本问题,pyTorch>=0.5, 改为loss_val.data即可