Commit 5d0d6d16 authored by Yuxin Wu's avatar Yuxin Wu

bump tensorpack version

parent 26b4ea44
...@@ -13,4 +13,4 @@ os.environ['TF_ENABLE_WINOGRAD_NONFUSED'] = '1' # issue#9339 ...@@ -13,4 +13,4 @@ os.environ['TF_ENABLE_WINOGRAD_NONFUSED'] = '1' # issue#9339
os.environ['TF_AUTOTUNE_THRESHOLD'] = '3' # use more warm-up os.environ['TF_AUTOTUNE_THRESHOLD'] = '3' # use more warm-up
os.environ['TF_AVGPOOL_USE_CUDNN'] = '1' # issue#8566 os.environ['TF_AVGPOOL_USE_CUDNN'] = '1' # issue#8566
__version__ = '0.2.0' __version__ = '0.3.0'
...@@ -147,8 +147,8 @@ def layer_register( ...@@ -147,8 +147,8 @@ def layer_register(
_register(func.__name__, wrapped_func) _register(func.__name__, wrapped_func)
return wrapped_func return wrapped_func
# need some special handling for sphinx to work with the arguments
if building_rtfd(): if building_rtfd():
# preserve argspec
from decorator import decorator from decorator import decorator
wrapper = decorator(wrapper) wrapper = decorator(wrapper)
......
...@@ -112,7 +112,6 @@ class TowerContext(object): ...@@ -112,7 +112,6 @@ class TowerContext(object):
for c in self._ctxs: for c in self._ctxs:
c.__enter__() c.__enter__()
# currently only check for predictor towers
if get_tf_version_number() >= 1.2: if get_tf_version_number() >= 1.2:
ns = tf.get_default_graph().get_name_scope() ns = tf.get_default_graph().get_name_scope()
assert ns == self._name, \ assert ns == self._name, \
......
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