qurator-spk/sbb_pixelwise_segmentation
1
0
Fork 0
mirror of https://github.com/qurator-spk/sbb_pixelwise_segmentation.git synced 2025-06-07 19:05:24 +02:00
Code Issues Releases Wiki Activity

Update main.py

Browse source
This commit is contained in:
vahidrezanezhad 2019-12-10 14:01:55 +01:00 • committed by GitHub
parent 3ac99b4e81
commit bb212daf0b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
train.py
View file

@ -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)