mirror of
https://github.com/qurator-spk/sbb_pixelwise_segmentation.git
synced 2025-06-07 19:05:24 +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(
|
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…
Add table
Add a link
Reference in a new issue