From d7e80786d889e5f8cfa97d9a6bdc43d23519d5b6 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Tue, 30 Apr 2019 18:26:48 +0200 Subject: [PATCH] Reset kernel cache after every batch Signed-off-by: Jim Martens --- src/twomartens/masterthesis/ssd.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/twomartens/masterthesis/ssd.py b/src/twomartens/masterthesis/ssd.py index 891db09..9cadfd7 100644 --- a/src/twomartens/masterthesis/ssd.py +++ b/src/twomartens/masterthesis/ssd.py @@ -200,6 +200,7 @@ def _predict_one_epoch(dataset: tf.data.Dataset, np.save(file, np.array(decoded_predictions_batch), allow_pickle=False, fix_imports=False) counter += 1 + tf.set_random_seed(1) epoch_end_time = time.time() per_epoch_time = epoch_end_time - epoch_start_time