pytorch错误：ValueError: too many dimensions ‘str‘***********num_samples should be a positive integeral

 调用以下函数时出错：ValueError: too many dimensions 'str'

train_u = torch.LongTensor(train_u)

检查数据发现，数据为：

{"readerID": 1, "ISBN": "9787121310928", "score": 5}
{"readerID": 5, "ISBN": "9787121310928", "score": 5}

ISBN为字符串形式，在读取数据加了一个强转，work

 

总结：LongTensor()功能是将arrary 转化为 tensor  ,list元素必须为数

 

num_samples should be a positive integeral value, but got num_samples=0

出错:

test_loader = torch.utils.data.DataLoader(testset, batch_size=args.test_batch_size, shuffle=True)

原因：

Set数据集为空：[]

测试训练集，划分时一个testSet 没有分到数据