定义device
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
数据和模型都要传给device
数据传给device
train_iterator, valid_iterator, test_iterator = BucketIterator.splits(
(train_data, valid_data, test_data),
batch_size = BATCH_SIZE,
device = device)
模型传给设备