fix bug in 552c2b3

tensorpack · Jan 12, 2020 · 3c14ff6 · 3c14ff6
1 parent ffd9eae
commit 3c14ff6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorpack/graph_builder/utils.py b/tensorpack/graph_builder/utils.py
@@ -25,7 +25,7 @@ def _replace_global_by_local(kwargs):
     if 'collections' in kwargs:
         collections = kwargs['collections']
     if not collections:
-        collections = set(tf.GraphKeys.GLOBAL_VARIABLES)
+        collections = {tf.GraphKeys.GLOBAL_VARIABLES}
     else:
         collections = set(collections.copy())
     collections.remove(tf.GraphKeys.GLOBAL_VARIABLES)