mirror of
https://github.com/qurator-spk/sbb_pixelwise_segmentation.git
synced 2025-06-08 11:20:48 +02:00
Update main.py
This commit is contained in:
parent
3ac99b4e81
commit
bb212daf0b
1 changed files with 1 additions and 1 deletions
2
train.py
2
train.py
|
@ -169,7 +169,7 @@ def run(n_classes,n_epochs,input_height,
|
|||
|
||||
model.fit_generator(
|
||||
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_steps=1,
|
||||
epochs=n_epochs)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue