(C++) Where ist the mnist source code, not header file?

RuFAI · July 30, 2019, 12:32pm

Hi, where can I find the source code that belongs to the mnist.h header file? I am interested in the specific implementation of the data loading procedure but as expected there is just the decleation of the used methods in the header file.

Would be glad to get some information

dongfangduoshou123 · July 30, 2019, 1:04pm

大哥在这呢。

github.com

pytorch/pytorch/blob/master/torch/csrc/api/src/data/datasets/mnist.cpp

#include <torch/data/datasets/mnist.h>

#include <torch/data/example.h>
#include <torch/types.h>

#include <c10/util/Exception.h>

#include <cstddef>
#include <fstream>
#include <string>
#include <vector>

namespace torch {
namespace data {
namespace datasets {
namespace {
constexpr uint32_t kTrainSize = 60000;
constexpr uint32_t kTestSize = 10000;
constexpr uint32_t kImageMagicNumber = 2051;
constexpr uint32_t kTargetMagicNumber = 2049;

This file has been truncated. show original