Skip to content

Commit

Permalink
Merge pull request #1220 from SalahAdDin/patch-1
Browse files Browse the repository at this point in the history
Update train.py
  • Loading branch information
de-vri-es authored Dec 13, 2019
2 parents 56c6998 + 050cdca commit a70c929
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion keras_retinanet/bin/train.py
Original file line number Diff line number Diff line change
Expand Up @@ -519,7 +519,7 @@ def main(args=None):
use_multiprocessing=args.multiprocessing,
max_queue_size=args.max_queue_size,
validation_steps = args.steps_for_validation,
validation_data=validation_generator
validation_data=validation_generator,
initial_epoch=args.initial_epoch
)

Expand Down

0 comments on commit a70c929

Please sign in to comment.