Mayank Kumar Pathak

Thanks for your reply can you please mention the server requirements to train the model. On Wed, Aug 12, 2020 at 1:05 PM Saransh Mehta wrote: > Hi @mayankpathaklumiq >...

RUN_THIS.ipynb # Create data generators to flow images from directory. This improves the speed of training test_datagen = tf.keras.preprocessing.image.ImageDataGenerator(rescale=1./255) gen = test_datagen.flow_from_directory( directory="Data/Dataset1000/Test/WM_Images/", target_size=(GENERATE_SQUARE, GENERATE_SQUARE), color_mode="rgb", batch_size=1, class_mode="input", shuffle=True)

also provide the proper documentation how to run the code for training as well as for prediction

> Hey @mayankpathaklumiq, were you able to solve this issue? yes