Following the https://github.com/goldsborough/examples/blob/cpp/cpp/mnist/mnist.cpp, I am trying to write my own program for training using libtorch.
However, I don’t find any documents about how to load my own dataset.
It seems that the C++ API is similar to Python API.
First, using torch::data::datasets to create a object of dataset.
Second, using torch::data::make_data_loader to create a pointer of loader.
But, I don’t know how to define custom dataset using torch::data::datasets. Does anyone can help me?