How to create a custom dataloader in c++

C++
1

Hi all follow developers,

I have been struggling to create a custom dataloader in c++ torch. I think I am missing some key concepts. Could someone help guide me to the right path?

I have both input and target. However, I am struggling to create a dataset to run: torch::data::make_data_loader

Thank you.

2

What kind of issues are you seeing? This code snippet shows a minimal example how to use make_data_loader.

1 Like