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

fix decorator bug

parent b5495059
#!/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
......
......@@ -6,7 +6,7 @@ import tensorflow as tf
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):
......
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