Fix handling of InvalidArgumentError in training loop

X-DeepSpeech: NOBUILD
2020-01-08 17:10:47 +00:00 · 2020-01-08 17:10:47 +00:00 · f6cd28ba2d
commit f6cd28ba2d
parent 1dfba839ea
1 changed files with 3 additions and 1 deletions
--- a/DeepSpeech.py
+++ b/DeepSpeech.py
@ -602,8 +602,10 @@ def train():
                                    feed_dict=feed_dict)
                except tf.errors.InvalidArgumentError as err:
                    if FLAGS.augmentation_sparse_warp:
-                        log_info("skip sparse warp error: {}".format(err))
+                        log_info("Ignoring sparse warp error: {}".format(err))
                        continue
+                    else:
+                        raise
                except tf.errors.OutOfRangeError:
                    break