如何无法使MNIST数据集Pytorch C＆＃x2B;＆＃x2B;

Question

我正在尝试跟随此但是我需要在0到255之间加载MNIST数据集。 ？

我的代码是：

int main(int argc, char* argv[]) {

  const int64_t batch_size = 1;

  // MNIST Dataset
  auto train_dataset = torch::data::datasets::MNIST("./mnist")
      .map(torch::data::transforms::Stack<>());

  // Number of samples in the training set
  auto num_train_samples = train_dataset.size().value();

  cout << "Number of training samples: " << num_train_samples << endl;
  
  // Data loaders
  auto train_loader = torch::data::make_data_loader<torch::data::samplers::RandomSampler>(
  std::move(train_dataset), batch_size);

  for (auto& batch : *train_loader) {
    auto data = batch.data.view({batch_size, -1}).to(device);
    auto record = data[0].clone();
    cout << "Max value: " << max(record) << endl;
    cout << "Min value: " << max(record) << endl;
    break;
  }
}

我下载的MNIST数据集是原来的代码，来自 site 。

预先感谢您的帮助。

Answer 1

我已经查看了源文件，看来Pytorch MNIST数据集类别执行255的部门以在[0,1]范围内返回张量。因此，您将必须将批次乘以255。

归一化转换不是罪魁祸首。它用于改变数据的平均值和差异