Can I use imagedatagenerator in keras to train a pytorch cnn module?? If not is there any other keyword which does the same function in pytorch?
imagedatagenerator