no need to control order as tensor can belong to multiple collections…

… now (aws#49)
atqy · Jul 29, 2019 · 3187f6a · 3187f6a
1 parent 0e262a2
commit 3187f6a
Showing 1 changed file with 0 additions and 7 deletions.
diff --git a/tornasole_core/save_manager.py b/tornasole_core/save_manager.py
@@ -27,13 +27,6 @@ def __init__(self, collection_manager, include_collections_names,
     self.when_nan_tensors = {}
 
   def prepare(self):
-    # below is to control the order
-    # in which these collections appear in save_collections
-    for cname in ['weights', 'gradients', 'bias', 'optimizer_variables']:
-      if self._should_collection_be_saved(cname):
-        self.save_collections.append(self.collection_manager.get(cname))
-
-    # adding other collections to save_collections
     for c_name, c in self.collection_manager.get_collections().items():
       if self._should_collection_be_saved(c_name) \
         and c not in self.save_collections: