how to modify the train set and use the code

[parth-shettiwar]

Hey I am parth from India.
I saw your code and wanted to use it in my project.
Need to implement it on a custom dataset.
Please give supporting details or your contact for further interaction.
Thanks

[minggli]

Hi Parth,

you only need to modify pipeline so it returns "mnist_batch_iter" or equivalent to feed into computation graph. Hope it helps.