fix decorator bug

tensorpack · Apr 26, 2016 · 13f76d5 · 13f76d5
1 parent b549505
commit 13f76d5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/examples/ResNet/cifar10-resnet.py b/examples/ResNet/cifar10-resnet.py
@@ -1,6 +1,6 @@
 #!/usr/bin/env python
 # -*- coding: UTF-8 -*-
-# File: cifar10_resnet.py
+# File: cifar10-resnet.py
 # Author: Yuxin Wu <ppwwyyxx@gmail.com>
 
 import tensorflow as tf

diff --git a/tensorpack/models/_common.py b/tensorpack/models/_common.py
@@ -6,7 +6,7 @@
 from functools import wraps
 import six
 import copy
-import decorator
+#from decorator import decorator
 
 from ..tfutils import *
 from ..tfutils.modelutils import *
@@ -24,7 +24,7 @@ def layer_register(summary_activation=False, log_shape=True):
         Can be overriden when creating the layer.
     :param log_shape: log input/output shape of this layer
     """
-    @decorator.decorator
+    #@decorator only enable me when building docs.
     def wrapper(func):
         @wraps(func)
         def wrapped_func(*args, **kwargs):