最近在学习pytorch时,想在本地跑一些代码,但是速度太慢了,于是便根据教程更换了数据集大小,代码如下:
train_set.train_data = train_set.train_data[0:5000]
train_set.train_labels = train_set.train_labels[0:5000]
test_set.test_data = test_set.test_data[
最近在学习pytorch时,想在本地跑一些代码,但是速度太慢了,于是便根据教程更换了数据集大小,代码如下:
train_set.train_data = train_set.train_data[0:5000]
train_set.train_labels = train_set.train_labels[0:5000]
test_set.test_data = test_set.test_data[