Skip to content

S
seminar-breakout
Commit 3c14ff66 authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

fix bug in 552c2b3b

parent ffd9eaeb
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tensorpack/graph_builder/utils.py
View file @ 3c14ff66
...@@ -25,7 +25,7 @@ def _replace_global_by_local(kwargs): ...@@ -25,7 +25,7 @@ def _replace_global_by_local(kwargs):
if 'collections' in kwargs: if 'collections' in kwargs:
collections = kwargs['collections'] collections = kwargs['collections']
if not collections: if not collections:
collections = set(tf.GraphKeys.GLOBAL_VARIABLES) collections = {tf.GraphKeys.GLOBAL_VARIABLES}
else: else:
collections = set(collections.copy()) collections = set(collections.copy())
collections.remove(tf.GraphKeys.GLOBAL_VARIABLES) collections.remove(tf.GraphKeys.GLOBAL_VARIABLES)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment