pytorch中DataLoader读不出数据但是又没有报错的情况

今天学习pytorch,数据预处理的时候,将dataset的数据打包之,用dataloader进行封装后,进行测试,但是却一直读不出数据。输出的代码如下所示:

train_random = get_batch(random_embedding.train_s1_matrix, random_embedding.train_s2_matrix,
                             random_embedding.train_y,batch_size)
for i,j,k in train_random:
    print(i)
    print(j)
    print(k)
    break

上面的i,j是x,k是标签,一直读不到数据,如下:
pytorch中DataLoader读不出数据但是又没有报错的情况_第1张图片
然后debug发现dataset能正常读取,折腾了半天发现问题是:
我的 总数据 的数值,在此记录一下,虽然感觉是比较小的问题,但是大家在代码调试的时候可能也会先让程序读取小部分的数据看看输出结果,希望大家可以避开这个小问题。

你可能感兴趣的:(pytoch学习,pytorch,python,深度学习)