Hi, ALL
I have a question about the conversion of iter() option for make_data_loader in conversion to C++
the python code is:
unlabel_loader2 = DataLoader(self.unlabeled, batch_size = self.args.batch_size, shuffle=True, drop_last=True, num_workers = 4).iter()
In C++ code,
I write it as:
auto unlabel_loader2 = torch::data::make_data_loader(
unlabeled.map(torch::data::transforms::Normalize<>({0.5, 0.5, 0.5}, {0.5, 0.5, 0.5}))
.map(torch::data::transforms::Stack<>()),
torch::data::DataLoaderOptions().batch_size(args.batch_size).workers(4).drop_last(
true))
But I donot have the option of iter in C++
Any suggestion is welcome.
Thanks.