Make sets of tuples

Signed-off-by: Jim Martens <github@2martens.de>
2019-05-01 12:47:42 +02:00
parent 0392ea164f
commit dc0a4ab59b
1 changed files with 2 additions and 2 deletions
--- a/src/twomartens/masterthesis/ssd.py
+++ b/src/twomartens/masterthesis/ssd.py
@ -219,9 +219,9 @@ def _predict_one_epoch(dataset: tf.data.Dataset,
        all_objects = muppy.get_objects()
        all_lists = muppy.filter(all_objects, Type=list)
        if lists is None:
-            lists = set(all_lists)
+            lists = set(map(tuple, all_lists))
        else:
-            all_lists = set(all_lists)
+            all_lists = set(map(tuple, all_lists))
            print(all_lists.difference(lists))
    
    epoch_end_time = time.time()