Update main.py

pull/2/head
vahidrezanezhad 5 years ago committed by GitHub
parent 3ac99b4e81
commit bb212daf0b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -169,7 +169,7 @@ def run(n_classes,n_epochs,input_height,
model.fit_generator( model.fit_generator(
train_gen, train_gen,
steps_per_epoch=int(len(os.listdir(dir_flow_train_imgs))/n_batch), steps_per_epoch=int(len(os.listdir(dir_flow_train_imgs))/n_batch)-1,
validation_data=val_gen, validation_data=val_gen,
validation_steps=1, validation_steps=1,
epochs=n_epochs) epochs=n_epochs)

Loading…
Cancel
Save