Reset TF graph

pawel-czyz · Jun 2, 2018 · fa2b837 · fa2b837
1 parent 20bc347
commit fa2b837
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/cce/calculate_weighted_loss.py b/cce/calculate_weighted_loss.py
@@ -24,6 +24,9 @@ def weight_loss(neighb_count: np.array, labels: np.array, weights: list) -> floa
     float
         calculated loss
     """
+    # reset graph before each run
+    tf.reset_default_graph()
+
     num_data, num_labels = neighb_count.shape
     label_counts = np.zeros([num_labels])
     for label, count in Counter(labels).most_common():

diff --git a/cce/optimize_weights.py b/cce/optimize_weights.py
@@ -21,6 +21,9 @@ def weight_optimizer(neighb_count, labels) -> (float, list):
     list
         weight for each label
     """
+    # reset graph before each run
+    tf.reset_default_graph()
+
     num_data, num_labels = neighb_count.shape
     label_counts = np.zeros([num_labels])
     for label, count in Counter(labels).most_common():