Commit 13f76d5c authored by Yuxin Wu's avatar Yuxin Wu

fix decorator bug

parent b5495059
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: cifar10_resnet.py # File: cifar10-resnet.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com> # Author: Yuxin Wu <ppwwyyxx@gmail.com>
import tensorflow as tf import tensorflow as tf
......
...@@ -6,7 +6,7 @@ import tensorflow as tf ...@@ -6,7 +6,7 @@ import tensorflow as tf
from functools import wraps from functools import wraps
import six import six
import copy import copy
import decorator #from decorator import decorator
from ..tfutils import * from ..tfutils import *
from ..tfutils.modelutils import * from ..tfutils.modelutils import *
...@@ -24,7 +24,7 @@ def layer_register(summary_activation=False, log_shape=True): ...@@ -24,7 +24,7 @@ def layer_register(summary_activation=False, log_shape=True):
Can be overriden when creating the layer. Can be overriden when creating the layer.
:param log_shape: log input/output shape of this layer :param log_shape: log input/output shape of this layer
""" """
@decorator.decorator #@decorator only enable me when building docs.
def wrapper(func): def wrapper(func):
@wraps(func) @wraps(func)
def wrapped_func(*args, **kwargs): def wrapped_func(*args, **kwargs):
......
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